pytorch-template
pytorch-template copied to clipboard
Minor fixes to `Trainer` and `BaseTrainer` Initializers
self.metric_ftnswas previously assigned inBaseTrainer.__init__, but there are no uses ofself.metric_ftnsin BaseTrainer. What's more, it seems that tracking metrics is not a responsibility of BaseTrainer. Thus, this has been removed from BaseTrainer andself.metric_ftnsis now assigned inTrainer.__init__.self.configwas previously assigned in bothBaseTrainer.__init__andTrainer.__init__. It has been removed fromTrainer.