Open leonardoaraujosantos opened 6 years ago
Thanks, I will check that now
Hi @longcw now it works, here are the changes
leoara01@linuxdev:~/work/yolo2-pytorch$ git diff train.py
diff --git a/train.py b/train.py
index 8c33cb4..51ba809 100644
--- a/train.py
+++ b/train.py
@@ -75,10 +75,10 @@ for step in range(start_epoch * imdb.batch_per_epoch,
# backward
loss = net.loss
- bbox_loss += net.bbox_loss.data.cpu().numpy()[0]
- iou_loss += net.iou_loss.data.cpu().numpy()[0]
- cls_loss += net.cls_loss.data.cpu().numpy()[0]
- train_loss += loss.data.cpu().numpy()[0]
+ bbox_loss += net.bbox_loss.data.cpu().item()
+ iou_loss += net.iou_loss.data.cpu().item()
+ cls_loss += net.cls_loss.data.cpu().item()
+ train_loss += loss.data.cpu().item()
optimizer.zero_grad()
loss.backward()
optimizer.step()
Would you like that I create a Pull request for the issue?
Hi, I've cloned the code today and try the training and got the following error
I'm using pytorch 0.4