modify error in textvdcnn

topliftarm · Sep 18, 2019 · ce61b90 · ce61b90
1 parent 5372f6a
commit ce61b90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/classification/textvdcnn.py b/model/classification/textvdcnn.py
@@ -78,7 +78,7 @@ def get_parameter_optimizer_dict(self):
             params.append({'params': self.batch_norms[i].parameters()})
         params.append({'params': self.linear1.parameters()})
         params.append({'params': self.linear2.parameters()})
-        params.append({'params': self.linear3.parameters()})
+        params.append({'params': self.linear.parameters()})
         return params
 
     def update_lr(self, optimizer, epoch):