distributed_training/src_dir/main_trainer.py (1 line):
	- line 522: # TODO:  Change timings to mirror train().