Steven Lee
Steven Lee
How to run the code on multi GPUs?
## When i run visualization_vit.ipynb, i can not visualize the features 
Fix typo
Traceback (most recent call last): File "train.py", line 93, in trainer[dataset_name](args, net, snapshot_path) File "/home/lichangyong/Code/TransUNet/trainer.py", line 57, in trainer_synapse loss_dice = dice_loss(outputs, label_batch, softmax=True) File "/home/lichangyong/.pyenv/versions/pytorch_gpu/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl...