手机浏览器扫描二维码访问
#optimizer
optimizer=torch.optiAdam(net.parameters(),lr=lr)
#optimizer=torch.optiSGD(net.parameters(),lr=lr,momentum=0.9,weight_decay=5e-4)
scheduler=torch.optilr_scheduler.StepLR(optimizer,step_size=5,gamma=0.9)
ifnotos.path.exists(“logCNN“):
os.mkdir(“logCNN“)
writer=tensorboardX.SummaryWriter(“logCNN“)
forepochinrange(epoch_num):
train_sum_loss=0
train_sum_correct=0
train_sum_fp=0
train_sum_fn=0
train_sum_tp=0
train_sum_tn=0
fori,datainenumerate(trainDataLoader):
net.train()
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)
loss=loss_func(outputs,labels)
optimizer.zero_grad()
loss.backward()
optimizer.step()
_,pred=torch.max(outputs.data,dim=1)
acc=pred.eq(labels.data).cpu().sum()
one=torch.ones_like(labels)
zero=torch.zeros_like(labels)
tn=((labels==zero)*(pred==zero)).sum()
tp=((labels==one)*(pred==one)).sum()
fp=((labels==zero)*(pred==one)).sum()
fn=((labels==one)*(pred==zero)).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_loss=train_sum_loss*1.0len(trainDataLoader)
禁锢之日 高手达人的奇妙爱情 御兽:我能增强御兽血脉 李辰赵清澜赵蕊 倾月班 奸情曝光之后 剑影江湖传说 有樗 娇娇贵女红了眼,摄政王连夜哄 全村亿万富翁,你管这叫贫困村? 我的宗门有点怪! 穿越七零,勾的冷面军少心慌慌 七零之我靠纸扎手艺混的风生水起 副本存活率0?我直接速通! 我见南山 穿进甜文后成了霸总的恶毒前女友 和腹黑三叔闪婚后真香了 摆摊卖水果 我在修仙界养猪 贫僧法海,万世之尊
万众瞩目之下,楚浩扔出一柄剑这轩辕剑你拿好,以后别在我面前装逼。这天,这地,这沧海,这宇宙,谁都无法阻止我。ps看完了?新书搜索从诡秘复苏开始不当人推荐票刷起来,让我们再次征战。...
一个热爱网络游戏的痴孩子,二不垃及的真神祝愿下进入了游戏的世界。。。。。。...
...
关于抢救大明朝朱慈烺此贼比汉奸还奸,比鞑子还凶,比额李自成还能蛊惑人心!闯王李自成立马九宫山,遥望东南,感慨万千。慈烺此子忤逆不孝,奸诈凶残,简直是曹操再世,司马复生,让他当了皇帝,全天下的...
一个浑浑噩噩的少年,在阳台吹风不小心掉了下去,死过一次的他,决定开始改变,故事从这里开始,他就是林浩...
...