千千小说网

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

告密者的下场(第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?我直接速通!  穿越七零,勾的冷面军少心慌慌  奸情曝光之后  我在修仙界养猪  七零之我靠纸扎手艺混的风生水起  穿进甜文后成了霸总的恶毒前女友  摆摊卖水果  高手达人的奇妙爱情  我见南山  

热门小说推荐
大话之神

大话之神

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

凌天至尊

凌天至尊

最强系统,我就是最强!还有谁?叶风看着众多的天骄,脸色淡定无比!获得最强系统,经验可复制对方的功法神通,可升级功法神通品阶无所不能,唯有最强!碾压苍穹,打爆世间一切不服者!...

我和大圣是兄弟

我和大圣是兄弟

王虎穿越了,而且悲催的成了五指山下的一只老虎。我去,这是要做猴哥虎皮裙的节奏?王虎表示不服。作为一只21世纪穿越来的新时代老虎,怎么着也要和猴哥拜把子,做兄弟啊!此时此刻齐天大圣孙悟空被压五行山马上就满五百年,再有十年,波澜壮阔,影响三界格局的西天取经之旅就要开始,看王虎如何在其中搅动三界风云,与猴哥一起再掀万...

武林店小二

武林店小二

江湖日报讯肯麦郎连锁客栈享誉大明各府,其总部却是京城一家名为来福的小客栈。来福客栈在江湖上大名鼎鼎,即便费用高昂,上到各派掌门下到江湖游侠,都挤破脑袋想去来福客栈吃顿饭。记者有幸请到武林盟主,揭开来福客栈的秘密!来福客栈日常一幕少林方丈,你怎么吃饭不给钱啊?偶弥陀佛,出家人身无分文,这顿饭可否算作化缘?不行!武当掌门没钱吃饭,还在后院洗碗呢!你若不给钱,就去洗茅房!来福客栈日常二幕丐帮长老,瞧你样子就没钱吃饭,你来客栈干啥?听闻来福客栈可以拿东西抵押,我这里有本上乘的秘...

每日热搜小说推荐