mirrors/mmselfsup
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmselfsup.git synced 2025-06-03 14:59:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
xieenze 2020-09-09 16:56:55 +08:00
parent 4cd6eb6cba
commit d2284b373f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
openselfsup/apis/train.py
View File

@ -248,9 +248,12 @@ def _non_dist_train(model,
seed=cfg.seed, seed=cfg.seed,
drop_last=getattr(cfg.data, 'drop_last', False)) for ds in dataset drop_last=getattr(cfg.data, 'drop_last', False)) for ds in dataset
] ]
if 'use_fp16' in cfg and cfg.use_fp16 == True:
raise NotImplementedError('apex do not support non_dist_trian!')
# put model on gpus # put model on gpus
model = MMDataParallel(model, device_ids=range(cfg.gpus)).cuda() model = MMDataParallel(model, device_ids=range(cfg.gpus)).cuda()
# build runner # build runner
optimizer = build_optimizer(model, cfg.optimizer) optimizer = build_optimizer(model, cfg.optimizer)
runner = Runner( runner = Runner(