search

RL4M / PCRLv2

An official implementation of PCRLv2 (pre-training and fine-tuning code are included).
MIT License
113 stars 8 forks source link

Error when finetuning on Brats #12

Closed joangog closed 10 months ago

joangog commented 10 months ago

I get the following error when running the command:

python main.py --data /mnt/5C5C25FB5C25D116/data/BraTS2018 --model pcrlv2 --phase finetune --lr 1e-4 --output ./brats_finetune_weight --weight ./pretrained_weight/simance_multi_crop_luna_pretask_1.0_240.pt --n brats --d 3 --gpus 0,1,2,3 --b 4 --ratio 1.0

image

joangog commented 10 months ago

Downgrading to python 3.6 solved it