From 9766ef0f3a539be7ee68bb93918f25a3298afe39 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 14 May 2018 19:15:53 +0200 Subject: stub in primer --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model.py') diff --git a/model.py b/model.py index d3716d1..567e533 100644 --- a/model.py +++ b/model.py @@ -242,7 +242,7 @@ class Generator(Runner): super().__init__(model) self.cuda = cuda - def __call__(self, n_seqs, seq_len): + def __call__(self, n_seqs, seq_len, primer): # generation doesn't work with CUDNN for some reason torch.backends.cudnn.enabled = False -- cgit v1.2.3-70-g09d2