Train the model for one epoch.
(args, trainer, task, epoch_itr)
source not stored for this graph (policy: none)
no test coverage detected