From 333bacd26376250d8606d737f01361853527f451 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 12 Jan 2021 23:16:55 +0800 Subject: Bug fixes 1. Replace inplace Leaky ReLU in auto-encoder classifier with non-inplace one 2. Replace rank number with get_ordinal method in xmp --- models/auto_encoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'models/auto_encoder.py') diff --git a/models/auto_encoder.py b/models/auto_encoder.py index 64c52e3..0c247f1 100644 --- a/models/auto_encoder.py +++ b/models/auto_encoder.py @@ -126,7 +126,7 @@ class AutoEncoder(nn.Module): f_c_dim = embedding_dims[1] self.classifier = nn.Sequential( - nn.LeakyReLU(0.2, inplace=True), + nn.LeakyReLU(0.2), BasicLinear(f_c_dim, num_class) ) -- cgit v1.2.3