Alexander Tong
Alexander Tong
Sorry, I was hacking for the deadline and left in the custom matplotlib ticklabels. Just fixed this in `/lib/viz_scrna.py` line 358. That is the function that generates the density plots...
By default it is set up to use a gpu if available. Please check that one is available with `torch.cuda.isavailable()`. There is a `--gpu` flag for specifying gpu and a...
Things were broken in pytorch 1.4 specifically for the density regularization on GPU anything 1.5 and up should be fine.
you should be able to load weights from a checkpoint like this, maybe inserting this around line 450 in main.py (and obviously with a correct path) would do the trick...
LGTM. Thanks for the contribution @ImahnShekhzadeh