NameError: name 'BNOptimizer' is not defined

[Worromots]

在稀疏训练时，NameError: name 'BNOptimizer' is not defined，请问这个怎么解决

[violet17]

@TSlYY 作者还没整理好prune_util.py. 着急可以看我的仓库。。

[Worromots]

@violet17 已经下载过prune_util.py了啊，还是报错

[molyswu]

Get dataloader

dataset = ListDataset(train_path, augment=True, multiscale=opt.multiscale_training)
dataloader = torch.utils.data.DataLoader(
    dataset,
    batch_size=opt.batch_size,
    shuffle=True,
    num_workers=opt.n_cpu,
    pin_memory=True,
    collate_fn=dataset.collate_fn
)

optimizer = torch.optim.SGD(model.parameters(), lr=opt.lr, momentum=0.9)

---

for batch_i, (_, imgs, targets) in enumerate(dataloader):
batches_done = len(dataloader) * epoch + batch_i

        imgs = imgs.to(device)
        targets = targets.to(device)

        loss, outputs = model(imgs, targets)

        optimizer.zero_grad()
        loss.backward()

        updateBN(sr_flag, model.module_list, opt.s, prune_idx)
        #BNOptimizer.updateBN(sr_flag, model.module_list, opt.s, prune_idx)

        optimizer.step()

[Worromots]

谢谢你，问题解决了 ---原始邮件--- 发件人:"molyswu"<notifications@github.com> 发送时间:2019年8月1日 星期四 下午7:13 收件人:"Lam1360/YOLOv3-model-pruning"<YOLOv3-model-pruning@noreply.github.com> 抄送:"TSlYY"<1285317663@qq.com>;"Mention"<mention@noreply.github.com> 主题:Re: [Lam1360/YOLOv3-model-pruning] NameError: name 'BNOptimizer' isnot defined (#33) Get dataloader dataset = ListDataset(train_path, augment=True, multiscale=opt.multiscale_training) dataloader = torch.utils.data.DataLoader( dataset, batch_size=opt.batch_size, shuffle=True, num_workers=opt.n_cpu, pin_memory=True, collate_fn=dataset.collate_fn ) optimizer = torch.optim.SGD(model.parameters(), lr=opt.lr, momentum=0.9) &mdash; You are receiving this because you were mentioned. Reply to this email directly,view it on GitHub, ormute the thread.