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

fix: -> DistributedBatchSampler

上级 e4327c86
...@@ -15,6 +15,13 @@ Global: ...@@ -15,6 +15,13 @@ Global:
image_shape: [*image_channel, 224, 224] image_shape: [*image_channel, 224, 224]
save_inference_dir: ./inference save_inference_dir: ./inference
# mixed precision training
AMP:
scale_loss: 128.0
use_dynamic_loss_scaling: True
# O2: pure fp16
level: O2
# model architecture # model architecture
Arch: Arch:
name: SE_ResNeXt101_32x4d name: SE_ResNeXt101_32x4d
...@@ -32,13 +39,6 @@ Loss: ...@@ -32,13 +39,6 @@ Loss:
- CELoss: - CELoss:
weight: 1.0 weight: 1.0
# mixed precision training
AMP:
scale_loss: 128.0
use_dynamic_loss_scaling: True
# O2: pure fp16
level: O2
Optimizer: Optimizer:
name: Momentum name: Momentum
momentum: 0.9 momentum: 0.9
...@@ -102,7 +102,7 @@ DataLoader: ...@@ -102,7 +102,7 @@ DataLoader:
output_fp16: True output_fp16: True
channel_num: *image_channel channel_num: *image_channel
sampler: sampler:
name: BatchSampler name: DistributedBatchSampler
batch_size: 64 batch_size: 64
drop_last: False drop_last: False
shuffle: False shuffle: False
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册