Merge pull request Tencent#20 from SannyZhou/master

modify error in textvdcnn
Shuriken13 · Sep 18, 2019 · 0c81897 · 0c81897
2 parents e3709b4 + ce61b90
commit 0c81897
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):