笔趣阁

手机浏览器扫描二维码访问

告密者的下场(第3页)

#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?我直接速通!  有樗  和腹黑三叔闪婚后真香了  李辰赵清澜赵蕊  摆摊卖水果  七零之我靠纸扎手艺混的风生水起  剑影江湖传说  娇娇贵女红了眼,摄政王连夜哄  我在修仙界养猪  全村亿万富翁,你管这叫贫困村?  我的宗门有点怪!  高手达人的奇妙爱情  禁锢之日  

热门小说推荐
龙符

龙符

苍茫大地,未来变革,混乱之中,龙蛇并起,谁是真龙,谁又是蟒蛇?或是天地众生,皆可成龙?朝廷,江湖门派,世外仙道,千年世家,蛮族,魔神,妖族,上古巫道,千百势力,相互纠缠,因缘际会。...

总裁大人超给力

总裁大人超给力

嫁给我,我可以替你报仇。陆白,亚洲第一跨国集团帝晟集团总裁,商业界最可怕的男人。传闻他身后有着最庞大的金融帝国,身边从未有过什么女人,传说他是夏儿想,管他呢,安心地做她的总裁夫人虐虐渣最好不过了。只是婚后生活渐渐地不一样了,看着报纸上帝晟总裁的采访,安夏儿方了你你你什么意思,不是说好我们隐婚的么老...

绝色占卜师:爷,你挺住!

绝色占卜师:爷,你挺住!

听说她在占卜,他捧着手眼巴巴的就过来了爱卿,你给本君算算,今晚是本君睡了国师呢?还是国师睡了本君?她哆嗦了一下,一脚就踹了过去谁都不睡!她今晚就阉了你!!重生前,她是惊才绝艳的大占卜师,重生后,她还是上知天文下知地理的一品国师,可是,她算了两世,却没算到自己这一世会犯桃花国师大人,不好了,帝君来了!卧槽!她一下子就从八卦盘里站了起来他来干什么?他不干什么!那就好那就好!她狂抹一把额头上的冷汗。小太监欲哭无泪可他说了,今晚他夜观星象,是个鸾凤和鸣...

每日热搜小说推荐