手机浏览器扫描二维码访问
#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?我直接速通! 有樗 御兽:我能增强御兽血脉 剑影江湖传说 奸情曝光之后 穿越七零,勾的冷面军少心慌慌 穿进甜文后成了霸总的恶毒前女友 贫僧法海,万世之尊
嫁给我,我可以替你报仇。陆白,亚洲第一跨国集团帝晟集团总裁,商业界最可怕的男人。传闻他身后有着最庞大的金融帝国,身边从未有过什么女人,传说他是夏儿想,管他呢,安心地做她的总裁夫人虐虐渣最好不过了。只是婚后生活渐渐地不一样了,看着报纸上帝晟总裁的采访,安夏儿方了你你你什么意思,不是说好我们隐婚的么老...
...
...
一个转世失败的神农弟子,想过咸鱼般的田园生活?没机会了!不靠谱的神农,会让你体验到忙碌而充实的感觉。师父别闹,就算我病死饿死从悬崖跳下去,也不种田,更不吃你赏赐的美食真香啊!本人著有完本精品农家仙田,欢迎阅读。QQ群42993787...
这里不是春秋战国,也不是东汉末年!似曾相识的齐楚秦魏,截然不同的列国争雄!来自现代的灵魂,入主王侯之家,却命悬一线君王之路,前狼后虎,他将何去何从!大争之世,群雄并起,谁能主宰沉浮!魏无忌寡人这辈子只信刀剑说话,管你诸侯还是天子,不服的有种就站出来。本书群号327625454PS本书...
...