-
Notifications
You must be signed in to change notification settings - Fork 30
Open
Labels
bugSomething isn't workingSomething isn't working
Description
- Devolearn requires
torch==1.7, but I think it would also work with the latest versions of torch. Using the latets version of torch would make the installation much faster on google colab notebooks because the latest version of torch is already installed there. - The URL for the weights to the embryo segmentation model is most likely broken, throws an HTTP error.
- arg
mode = 'cpu'should be replaced withdevice = 'cpu' - optionally, we can also let the user decide the GPU ID like
device = 'cuda:0' - move the
run testscell to the bottom
Note that this is a beginner friendly issue, all the changes are to be done on this notebook. Feel free to reply to this if you're willing to take this up.
Mainakdeb
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't working