训练模型的剩余时间

import time
import datetime
prev_time = time.time() #epoch循环之外
# epoch训练循环之内
        batches_done = (epoch - start_iter + 1) * len(data_loader) + _ + 1
        batches_left = (args.train_epochs + 2 - start_iter) * len(data_loader) - batches_done
        time_left = datetime.timedelta(seconds=batches_left * (time.time() - prev_time))
        prev_time = time.time()
print('[Epoch %d/%d] [Batch %d/%d] [count %d] [G_loss %f] [time %s] ' %
              (epoch, args.train_epochs, _ + 1, len(data_loader), count, G_loss, time_left))