examples/poyo/train.py does not work without these hacks to bypass some errors. Can be fixed properly later, but I think we need a working setup right now.
Disable data leakage: Because new data preprocessing scripts don't add a "{split}_mask"
Deleting Depth_classes in capoyo, because this class does not exist in taxonomy/mice.py. Causes a compile error in poyo/train.py
Remove .cpu() in model weights logging: Some lightning features don't work well (eg. checkpointing) when a logged value with sync_dist=True is on the CPU
examples/poyo/train.pydoes not work without these hacks to bypass some errors. Can be fixed properly later, but I think we need a working setup right now.Depth_classesin capoyo, because this class does not exist intaxonomy/mice.py. Causes a compile error in poyo/train.py.cpu()in model weights logging: Some lightning features don't work well (eg. checkpointing) when a logged value withsync_dist=Trueis on the CPU