riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Minor changes

Browse Source
This commit is contained in:
piero 2019-11-27 15:27:49 -08:00 committed by Julien Chaumond
parent 7469d03b1c
commit 821de121e8
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/run_pplm_discrim_train.py
View File

@ -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):
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]