fix: -> DistributedBatchSampler
@@ -15,6 +15,13 @@ Global:
image_shape: [*image_channel, 224, 224]
save_inference_dir: ./inference
+# mixed precision training
+AMP:
+ scale_loss: 128.0
+ use_dynamic_loss_scaling: True
+ # O2: pure fp16
+ level: O2
+
# model architecture
Arch:
name: SE_ResNeXt101_32x4d
@@ -32,13 +39,6 @@ Loss:
- CELoss:
weight: 1.0
-# mixed precision training
-AMP:
- scale_loss: 128.0
- use_dynamic_loss_scaling: True
- # O2: pure fp16
- level: O2
-
Optimizer:
name: Momentum
momentum: 0.9
@@ -102,7 +102,7 @@ DataLoader:
output_fp16: True
channel_num: *image_channel
sampler:
- name: BatchSampler
+ name: DistributedBatchSampler
batch_size: 64
drop_last: False
shuffle: False
