Closed Stephen-Hao closed 3 years ago
use ddp for multi-gpu TResNet.
see https://github.com/Alibaba-MIIL/ImageNet21K/blob/main/train_single_label.py as an example for how to properly initilize multi-gpu distributed (
model = setup_distrib(model, args)
)
Originally posted by @mrT23 in https://github.com/Alibaba-MIIL/TResNet/issues/35#issuecomment-826610266
use ddp for multi-gpu TResNet.
see https://github.com/Alibaba-MIIL/ImageNet21K/blob/main/train_single_label.py as an example for how to properly initilize multi-gpu distributed (
setup distributed
)
Originally posted by @mrT23 in https://github.com/Alibaba-MIIL/TResNet/issues/35#issuecomment-826610266