diff options
Diffstat (limited to 'gantools/biggan.py')
| -rw-r--r-- | gantools/biggan.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/gantools/biggan.py b/gantools/biggan.py index 8a0a71d..0ed6bf7 100644 --- a/gantools/biggan.py +++ b/gantools/biggan.py @@ -6,7 +6,7 @@ from itertools import cycle MODULE_PATH = 'https://tfhub.dev/deepmind/biggan-512/2' -class BigGAN: +class BigGAN(object): def __init__(self, module_path=MODULE_PATH): tf.reset_default_graph() print('Loading BigGAN module from:', module_path) |