search

OliverRensu / Shunted-Transformer

205 stars 20 forks source link

About distributed training #4

Open Dratlan opened 2 years ago

Dratlan commented 2 years ago

Exciting work. 👍I'm trying to run the dist_train.sh script you gave. And get the error below. I only have one GPU. Is this the error caused by this? Or is this only for distributed training? Is there another training method } 1O9TM5P$KC W%IN90HTN7 ~FBWW}(B86WLB2RNSR`273P Looking forward to your reply. Thank you.

mywebinfo65536 commented 1 year ago

Hi, you can see the detail of the dist_train.sh script, and then you can just run the main.py script with some parameter. have a try