告密者的下场

train_sum_fn+train_sum_tp)

writer.add_scalar(“trainloss“,train_loss,global_step=epoch)

writer.add_scalar(“traincorrect“,

train_correct,global_step=epoch)

writer.add_scalar(“trainprecision“,

train_precision,global_step=epoch)

writer.add_scalar(“trainrecall“,train_recall,global_step=epoch)

ifnotos.path.exists(“models_aug_CNN“):

os.mkdir(“models_aug_CNN“)

torch.save(net.state_dict(),“models_aug_CNN/{}.pth“.format(epoch+1))

scheduler.step()

sum_loss=0

sum_correct=0

test_sum_fp=0

test_sum_fn=0

test_sum_tp=0

test_sum_tn=0

fori,datainenumerate(testDataLoader):

net.eval()

inputs,labels=data

inputs=inputs.unsqueeze(1).to(torch.float32)

labels=labels.type(torch.LongTensor)

inputs,labels=inputs.to(device),labels.to(device)

outputs=net(inputs)

本章未完，请点击下一页继续阅读！ 第6页 / 共7页