Remove DataParallel from the model state before saving

2026-09-12 12:11:47 +08:00 · 2018-02-21 07:03:53 -08:00
parent 2c345b622d
commit 1bfd8f73e7
4 changed files with 25707 additions and 277 deletions
@@ -213,7 +213,7 @@ class Decoder(nn.Module):
        r (int): number of outputs per time step.
        eps (float): threshold for detecting the end of a sentence.
    """
-    def __init__(self, in_features, memory_dim, r, eps=0.2):
+    def __init__(self, in_features, memory_dim, r, eps=0.05):
        super(Decoder, self).__init__()
        self.max_decoder_steps = 200
        self.memory_dim = memory_dim