Documentation |
API |
Examples |
Citation
int main(int argc, char* argv[]) {
MyModel model;
using Trainer = dl::InferTrainer<MyModel>;
Trainer trainer(Trainer::Settings{
.createDataset = [] { return std::make_unique<MyDataset>(); },
.loss = dl::loss::mse,
.optimizer = std::make_unique<dl::optimizer::Adam>(),
.limitEpochs = 10,
});
trainer.fit(model);
trainer.test(model);
}
FetchContent_Declare(libdl GIT_REPOSITORY https://github.com/TheMrSheldon/libdl.git)
FetchContent_MakeAvailable(libdl)
target_link_libraries(<mytarget> PUBLIC libdl)