How to run in Apple Silicon Mac?

Open smjxpro opened this issue 2 years ago • 1 comments

I have an Apple Silicon device (MacBook Pro M1Pro). When trying to run python train_fusion_model.py --dataset_path 'datasets/msrs_train' --epochs 30 --cls_pretrained 'pretrained/best_cls.pth' --batch_size 128 --lr 0.001 --loss_weight '[3, 7, 50]' --save_path 'pretrained' it throws an error saying AssertionError: Torch not compiled with CUDA enabled

Dec 18 '23 05:12 smjxpro

I haven't tested it on any mac device, sorry.

Jan 22 '24 05:01 linklist2