e*predzero.sum
train_sum_fn+fn.item
train_sum_fp+fp.item
train_sum_tn+tn.item
train_sum_tp+tp.item
train_sum_loss+loss.item
train_sum_correct+acc.item
train_losstrain_sum_loss*1.0lentraindataloader
train_correcttrain_sum_correct*1.0lentraindataloaderbatch_size
train_precisiontrain_sum_tp*1.0train_sum_fp+train_sum_tp
train_recalltrain_sum_tp*1.0train_sum_fn+train_sum_tp
writer.add_scalar“trainloss“,train_loss,global_stepepoch
writer.add_scalar“traincorrect“,
train_correct,global_stepepoch
writer.add_scalar“trainprecision“,
train_precision,global_stepepoch
writer.add_scalar“trainrecall“,train_recall,global_stepepoch
“models_aug_cnn“:
os.mkdir“models_aug_cnn“
,“models_aug_cnn.pth“.formatepoch+1
scheduler.step
sum_loss0
sum_correct0
test_sum_fp0
test_sum_fn0
test_sum_tp0
test_sum_tn0