From 821de121e86574504ec648f76ccb924e38125b52 Mon Sep 17 00:00:00 2001
From: piero <piero@uber.com>
Date: Wed, 27 Nov 2019 15:27:49 -0800
Subject: [PATCH] Minor changes

---
 examples/run_pplm_discrim_train.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/examples/run_pplm_discrim_train.py b/examples/run_pplm_discrim_train.py
index 7f10e861a8..9438cbbac1 100644
--- a/examples/run_pplm_discrim_train.py
+++ b/examples/run_pplm_discrim_train.py
@@ -72,7 +72,6 @@ class Discriminator(torch.nn.Module):
     def train_custom(self):
         for param in self.encoder.parameters():
             param.requires_grad = False
-        pass
         self.classifier_head.train()
 
     def avg_representation(self, x):
@@ -122,7 +121,7 @@ def collate_fn(data):
         padded_sequences = torch.zeros(
             len(sequences),
             max(lengths)
-        ).long()  # padding index 0
+        ).long()  # padding value = 0
 
         for i, seq in enumerate(sequences):
             end = lengths[i]