From b87398674ae01f26e34709b32c7d4b2c306cc703 Mon Sep 17 00:00:00 2001 From: wogong Date: Mon, 28 Jan 2019 19:39:32 +0800 Subject: [PATCH] add default value for alpha --- models/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/model.py b/models/model.py index 7fd3f20..3d3b651 100644 --- a/models/model.py +++ b/models/model.py @@ -141,7 +141,7 @@ class SVHNmodel(nn.Module): nn.Linear(256, 2), ) - def forward(self, input_data, alpha): + def forward(self, input_data, alpha = 1.0): input_data = input_data.expand(input_data.data.shape[0], 3, 28, 28) feature = self.feature(input_data) feature = feature.view(-1, 64 * 4 * 4)