Train the ViT in 4 2080Ti

taoyang1122 / adapt-image-models

[ICLR'23] AIM: Adapting Image Models for Efficient Video Action Recognition

Apache License 2.0

278 stars 21 forks source link

Train the ViT in 4 2080Ti #34

Open wlsrick opened 1 year ago

wlsrick commented 1 year ago

Hello, I try to train the model with 4x2080Ti server, I use the command below, bash tools/dist_train.sh ./configs/recognition/vit/vitclip_large_k400.py 4 --test-last --validate --cfg-options work_dir=./work_dirs but it runs the error :

torch.distributed.elastic.multiprocessing.errors.ChildFailedError:

========= tools/train.py FAILED

How can I solve it? Thanks a lot~

taoyang1122 commented 1 year ago

Hi, could you please post the complete log here?