libdl

int main(int argc, char* argv[]) {

    MyModel model;


    auto conf = dl::TrainerConfBuilder<MyModel>()

                .setDataset<MyDataset>()

                .setOptimizer<dl::optim::GradientDescent>(model.parameters())

                .addObserver(dl::observers::limitEpochs(10))

                .addObserver(dl::observers::earlyStopping(3))

                .addObserver(dl::observers::consoleUI())

                .build();

    auto trainer = dl::Trainer(std::move(conf));

    trainer.fit(model, dl::lossAdapter(dl::loss::mse));

    trainer.test(model);

    return 0;

}

dl::TrainerConfBuilder
Definition trainer.hpp:46

dl::Trainer
Definition trainer.hpp:108

std::move
T move(T... args)