We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cc4645a commit df7b3e8Copy full SHA for df7b3e8
mmpose/apis/train.py
@@ -115,6 +115,7 @@ def train_model(model,
115
116
# get currently existing device type
117
device = get_device()
118
+ cfg.device = device
119
120
# put model on gpus
121
if distributed:
@@ -165,6 +166,8 @@ def train_model(model,
165
166
else:
167
# fp16 setting
168
fp16_cfg = cfg.get('fp16', None)
169
+ if fp16_cfg is None and cfg.get('device', None) == 'npu':
170
+ fp16_cfg = dict(loss_scale='dynamic')
171
if fp16_cfg is not None:
172
optimizer_config = Fp16OptimizerHook(
173
**cfg.optimizer_config, **fp16_cfg, distributed=distributed)
0 commit comments