thank you, but where should i put the training_start_time?
This is how i normally sum up my train time
dur = []
for epoch in range(1, 100):
    t0 = time.time()
    train(epoch)
    train_acc = test(train_loader)
    test_acc = test(test_loader)
  

    
    dur.append(time.time() - t0)

    print('Epoch: {:03d}, Train: {:.4f}, Test: {:.4f},Time(s) {:.4f}'.format(epoch,train_acc,test_acc,np.mean(dur)))

when i use your code, it isn’t working
image1536×348 15 KB