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"[email protected] 发送时间:2019年8月1日 星期四 下午7:13 收件人:"Lam1360/YOLOv3-model-pruning"[email protected] 抄送:"TSlYY"[email protected];"Mention"[email protected] 主题: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) — You are receiving this because you were mentioned. Reply to this email directly,view it on GitHub, ormute the thread.