search

victoresque / pytorch-template

PyTorch deep learning projects made easy.
MIT License
4.75k stars 1.09k forks source link

calling Trainer with len_epoch parameter #71

Closed doitrightnow1 closed 4 years ago

doitrightnow1 commented 4 years ago

Hi,

There seems to be an issue when setting len_epoch parameter and calling an instance of Trainer in this way: trainer = Trainer(model, criterion, metrics, optimizer, config=config, data_loader=data_loader, valid_data_loader=valid_data_loader, lr_scheduler=lr_scheduler, len_epoch=config['trainer']['steps_per_epoch'])