From e470c3aa1b5f457de9948ae1c4733ed09155f8e7 Mon Sep 17 00:00:00 2001
From: RangiLyu <lyuchqi@gmail.com>
Date: Tue, 21 Jun 2022 14:35:22 +0800
Subject: [PATCH] [Fix]: fix SWA in pytorch 1.6 (#312)
---
mmengine/model/averaged_model.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mmengine/model/averaged_model.py b/mmengine/model/averaged_model.py
index ab9ab9c0..e2f75e99 100644
--- a/mmengine/model/averaged_model.py
+++ b/mmengine/model/averaged_model.py
@@ -129,7 +129,7 @@ class StochasticWeightAverage(BaseAveragedModel):
"""
averaged_param.add_(
source_param - averaged_param,
- alpha=1 / (steps // self.interval + 1))
+ alpha=1 / float(steps // self.interval + 1))
@MODELS.register_module()
--
GitLab