epoch tot_tloss gw_tloss tot_vloss gw_vloss train_acc valid_acc roc_auc 0 0.6946 0.6946 0.6952 0.6952 0.5078 0.4865 0.5045