看书阁

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

告密者的下场(第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?我直接速通!  七零之我靠纸扎手艺混的风生水起  

热门小说推荐
九龙吞珠

九龙吞珠

一张从始皇帝皇宫流传出的长生不老药地图,解开不死不灭之秘。一代名将,将守,从万人敌,到无人敌的重生之路!九龙吞珠读者交流群721466643)...

大话之神

大话之神

一个热爱网络游戏的痴孩子,二不垃及的真神祝愿下进入了游戏的世界。。。。。。...

神农别闹

神农别闹

一个转世失败的神农弟子,想过咸鱼般的田园生活?没机会了!不靠谱的神农,会让你体验到忙碌而充实的感觉。师父别闹,就算我病死饿死从悬崖跳下去,也不种田,更不吃你赏赐的美食真香啊!本人著有完本精品农家仙田,欢迎阅读。QQ群42993787...

极品捉妖系统

极品捉妖系统

万众瞩目之下,楚浩扔出一柄剑这轩辕剑你拿好,以后别在我面前装逼。这天,这地,这沧海,这宇宙,谁都无法阻止我。ps看完了?新书搜索从诡秘复苏开始不当人推荐票刷起来,让我们再次征战。...

每日热搜小说推荐