bug fixed: dropout in FinalLayer should be 0

models/denoiser.py

@@ -42,7 +42,7 @@ def __init__(self, hparams):
             ]
         )
 
-        self.final_layer = FinalLayer(attn_dim, in_ch, dropout=dropout)
+        self.final_layer = FinalLayer(attn_dim, in_ch)
 
     def forward(self, x, cat, timesteps, key_padding_mask=None, graph_mask=None, attr_mask=None):
         # positional encoding for nodes and attributes