File tree 2 files changed +9
-8
lines changed
2 files changed +9
-8
lines changed Original file line number Diff line number Diff line change 406
406
" loop.set_description('total_loss = %.6f' % loss)\n " ,
407
407
" \n " ,
408
408
" # Save the losses for plotting\n " ,
409
- " chamfer_losses.append(loss_chamfer)\n " ,
410
- " edge_losses.append(loss_edge)\n " ,
411
- " normal_losses.append(loss_normal)\n " ,
412
- " laplacian_losses.append(loss_laplacian)\n " ,
409
+ " chamfer_losses.append(float( loss_chamfer.detach().cpu()) )\n " ,
410
+ " edge_losses.append(float( loss_edge.detach().cpu()) )\n " ,
411
+ " normal_losses.append(float( loss_normal.detach().cpu()) )\n " ,
412
+ " laplacian_losses.append(float( loss_laplacian.detach().cpu()) )\n " ,
413
413
" \n " ,
414
414
" # Plot mesh\n " ,
415
415
" if i % plot_period == 0:\n " ,
Original file line number Diff line number Diff line change 46
46
"id" : " okLalbR_g7NS"
47
47
},
48
48
"source" : [
49
- " Ensure `torch` and `torchvision` are installed. If `pytorch3d` is not installed, install it using the following cell:"
50
- ]
49
+ " Ensure `torch` and `torchvision` are installed. If `pytorch3d` is not installed, install it using the following cell:"
50
+ ]
51
51
},
52
52
{
53
53
"cell_type" : " code" ,
645
645
" sum_loss = torch.tensor(0.0, device=device)\n " ,
646
646
" for k, l in loss.items():\n " ,
647
647
" sum_loss += l * losses[k][\" weight\" ]\n " ,
648
- " losses[k][\" values\" ].append(l)\n " ,
648
+ " losses[k][\" values\" ].append(float(l.detach().cpu()))\n " ,
649
+ " \n " ,
649
650
" \n " ,
650
651
" # Print the losses\n " ,
651
652
" loop.set_description(\" total_loss = %.6f\" % sum_loss)\n " ,
829
830
" sum_loss = torch.tensor(0.0, device=device)\n " ,
830
831
" for k, l in loss.items():\n " ,
831
832
" sum_loss += l * losses[k][\" weight\" ]\n " ,
832
- " losses[k][\" values\" ].append(l )\n " ,
833
+ " losses[k][\" values\" ].append(float(l.detach().cpu()) )\n " ,
833
834
" \n " ,
834
835
" # Print the losses\n " ,
835
836
" loop.set_description(\" total_loss = %.6f\" % sum_loss)\n " ,
You can’t perform that action at this time.
0 commit comments