while using torch.save in main.py, torch not imported

M3DV / FracNet

[eBioMedicine] Deep-learning-assisted detection and segmentation of rib fractures from CT scans: Development and validation of FracNet

https://m3dv.github.io/FracNet/

Apache License 2.0

51 stars 25 forks source link

Closed agrawalshubham01 closed 2 years ago

kaimingkuang commented 3 years ago

Thanks! Can you delete the blank line between import torch and import torch.nn as nn? Then I will merge this pull request.

agrawalshubham01 commented 2 years ago

@kaimingkuang Updated

kaimingkuang commented 2 years ago

Merged. Thanks for your contribution.