Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "1efc208ff386fb6df56302c8f6f9484ddf93b92a"
Commit 821de121 authored by piero's avatar piero Committed by Julien Chaumond
Browse files

Minor changes

parent 7469d03b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
examples/run_pplm_discrim_train.py
View file @ 821de121
...@@ -72,7 +72,6 @@ class Discriminator(torch.nn.Module): ...@@ -72,7 +72,6 @@ class Discriminator(torch.nn.Module):
def train_custom(self): def train_custom(self):
for param in self.encoder.parameters(): for param in self.encoder.parameters():
param.requires_grad = False param.requires_grad = False
pass
self.classifier_head.train() self.classifier_head.train()
def avg_representation(self, x): def avg_representation(self, x):
...@@ -122,7 +121,7 @@ def collate_fn(data): ...@@ -122,7 +121,7 @@ def collate_fn(data):
padded_sequences = torch.zeros( padded_sequences = torch.zeros(
len(sequences), len(sequences),
max(lengths) max(lengths)
).long() # padding index 0 ).long() # padding value = 0
for i, seq in enumerate(sequences): for i, seq in enumerate(sequences):
end = lengths[i] end = lengths[i]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment