atria error
wshi8 opened this issue · 2 comments
Hello, thank you for all the wonderful project. I tried to run the atria task and got the following error, any advice?
python3 train3d.py --task atria --split all --bs 2 --maxiter 10000 --randscale 0.1 --net segtran --attractors 1024 --translayers 1
Traceback (most recent call last):
File "train3d.py", line 549, in
xyz_permute=args.xyz_permute)
TypeError: init() got an unexpected keyword argument 'mask_num_classes'
Sorry I'm busy working on the GAMMA challenge these days. Will check the code as soon as I have some free time.
Sorry for the error. It's because the atria dataset was no longer used in my later stage of development, and the relevant code becomes obsolete. I've updated datasets3d.py to make the interface of AtriaSet and MSDSet consistent with BratsSet. But I haven't actually run the code on the atria dataset. Could you please check out the updated code and see whether there are any errors?