search

facebookresearch / FAMBench

Benchmarks to capture important workloads.
Apache License 2.0
28 stars 23 forks source link

CvT update for torchrun #122

Closed pnunna93 closed 1 year ago

pnunna93 commented 1 year ago

CvT repo has changes to run with torchrun instead of torch.distributed.launch. This PR pins CvT to include those.

It also has changes to pull out END_EPOCH as a parameter user can control through their scripts.