Douzero_Resnet/train.py

9 lines
185 B
Python

import os
from douzero.dmc import parser, train
if __name__ == '__main__':
flags = parser.parse_args()
os.environ["CUDA_VISIBLE_DEVICES"] = flags.gpu_devices
train(flags)