提交 5f88903e 编写于 作者: G gaotingquan 提交者: Tingquan Gao

fix: -> DistributedBatchSampler

上级 e4327c86
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册