diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-05-14 19:15:53 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-05-14 19:15:53 +0200 |
| commit | 9766ef0f3a539be7ee68bb93918f25a3298afe39 (patch) | |
| tree | 442e2d0d86f6475d7b3d1ca0e288748e2a7f8f4d /model.py | |
| parent | 4167442627b1414ff8fdc86528812b46168c656b (diff) | |
stub in primer
Diffstat (limited to 'model.py')
| -rw-r--r-- | model.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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 |