From 67d6ee4d7a947456cab98b735c4d10b8378d954c Mon Sep 17 00:00:00 2001 From: Junwon Lee <63298243+cpprhtn@users.noreply.github.com> Date: Fri, 7 Oct 2022 20:43:36 +0900 Subject: [PATCH] Update pytorch_vision_wide_resnet.md --- pytorch_vision_wide_resnet.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_vision_wide_resnet.md b/pytorch_vision_wide_resnet.md index 2419a807..fbb27196 100644 --- a/pytorch_vision_wide_resnet.md +++ b/pytorch_vision_wide_resnet.md @@ -91,7 +91,7 @@ Otherwise the architecture is the same. Deeper ImageNet models with bottleneck block have increased number of channels in the inner 3x3 convolution. The `wide_resnet50_2` and `wide_resnet101_2` models were trained in FP16 with -mixed precision training using SGD with warm restarts. Checkpoints have weights in +mixed precision training using [SGD with warm restarts(SGDR)](https://arxiv.org/abs/1608.03983). Checkpoints have weights in half precision (except batch norm) for smaller size, and can be used in FP32 models too. | Model structure | Top-1 error | Top-5 error | # parameters |