[pplm] split classif head into its own file

This commit is contained in:
Julien Chaumond committed 2019-12-03 22:07:25 +00:00
1 parent 8101924a68
commit 7edb51f3a5
3 files changed
+20 -17

No files matched your search

+1 -1
View File
@@ -33,10 +33,10 @@ import torch.nn.functional as F
from torch.autograd import Variable
from tqdm import trange
from examples.run_pplm_discrim_train import ClassificationHead
from transformers import GPT2Tokenizer
from transformers.file_utils import cached_path
from transformers.modeling_gpt2 import GPT2LMHeadModel
from pplm_classification_head import ClassificationHead
PPLM_BOW = 1
PPLM_DISCRIM = 2