res optimizer DataPrefetcher OOM best_ap evaluate_and_save_model best after_train before_iter train_one_iter self.train_loaders self.train_loader before_train self.data train_loaders train_loader self.prefetcher prefe before_tra prefetcher progress epoch_progress before_task get_assignments adapt size gpu_mem_usage get_losses before id Citypersons true DEFINITION_ORDER