diff --git a/official/cv/ResNet/config/resnet50_imagenet2012_Boost_config.yaml b/official/cv/ResNet/config/resnet50_imagenet2012_Boost_config.yaml index c539e0a07e935ac7833e67c5f94fc07a1a87325e..ff6d2d8c1d98a8987485c6b65cc8bbe597ae9bbb 100644 --- a/official/cv/ResNet/config/resnet50_imagenet2012_Boost_config.yaml +++ b/official/cv/ResNet/config/resnet50_imagenet2012_Boost_config.yaml @@ -23,7 +23,7 @@ batch_size: 256 loss_scale: 1024 momentum: 0.85 weight_decay: 5.0e-5 -epoch_size: 60 +epoch_size: 90 start_epoch: 0 resume_ckpt: "" save_checkpoint: False