search

RL4M / PCRLv2

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

Lack of methods for importing brats data in DataGenerator code？ #8

Open liaochuanlin opened 1 year ago

liaochuanlin commented 1 year ago

pcrlv2_brats_finetune Traceback (most recent call last): File "/home/lcl/PCRLv2/main.py", line 46, in data_loader = get_dataloader(args) File "/home/lcl/PCRLv2/main.py", line 17, in get_dataloader dataloader = getattr(generator, loader_name)() AttributeError: 'DataGenerator' object has no attribute 'pcrlv2_brats_finetune'

joangog commented 12 months ago

You have to git checkout to the "finetune" branch of the repo.