s2.todevice
printnet
#loss
loss_funcnn.crossentropyloss
#optimizer
.parameters,lrlr
#.parameters,lrlr,momentum0.9,weight_decay5e-4
.steplroptimizer,step_size5,gamma0.9
“logcnn“:
os.mkdir“logcnn“
writertensorboardx.summarywriter“logcnn“
forepochinrangeepoch_num:
train_sum_loss0
train_sum_correct0
train_sum_fp0
train_sum_fn0
train_sum_tp0
train_sum_tn0
fori,datainenumeratetraindataloader:
net.train
inputs,labelsdata
.float32
inputs,labelsinputs.todevice,labels.todevice
outputsnetinputs
lossloss_funcoutputs,labels
optimizer.zero_grad
loss.backward
optimizer.step
_,,dim1
.cpu.sum
onetorch.ones_likelabels
zerotorch.zeros_likelabels
tnlabelszero*predzero.sum
tplabelsone*predone.sum
fplabelszero*predone.sum
fnlabelson