t-simcne

t-simcne copied to clipboard

How to load the trained model or checkpoints?

3

I have done the training, how can I load the .ckpt in lightning_logs/version_X/checkpoints/ ? So that I can re-plot the graph. I cannot find relevant API in document.

Air000

Unable to use multiple devices

6

It appears the number of devices is hardcoded: https://github.com/berenslab/t-simcne/blob/e2988f5cb2c86c321e5bad3bc1c968200097def0/tsimcne/tsimcne.py#L481 https://github.com/berenslab/t-simcne/blob/e2988f5cb2c86c321e5bad3bc1c968200097def0/tsimcne/tsimcne.py#L517 Given the contrastive learning task, it would be preferable to utilize more resources for training and allow the `devices` argument...

erikhagendorn

See issue #10

erikhagendorn