From 3a9989300a29d42086ace864eb5c4be724dec8fe Mon Sep 17 00:00:00 2001 From: kwotsin Date: Fri, 14 Aug 2020 14:27:16 +0800 Subject: [PATCH] fix sagan import --- setup.py | 2 +- torch_mimicry/__init__.py | 2 +- torch_mimicry/nets/sagan/__init__.py | 3 +++ 3 files changed, 5 insertions(+), 2 deletions(-) create mode 100644 torch_mimicry/nets/sagan/__init__.py diff --git a/setup.py b/setup.py index 75ed9ec..159c339 100644 --- a/setup.py +++ b/setup.py @@ -1,6 +1,6 @@ from setuptools import setup, find_packages -__version__ = '0.1.14' +__version__ = '0.1.16' url = 'https://github.com/kwotsin/mimicry' install_requires = [ diff --git a/torch_mimicry/__init__.py b/torch_mimicry/__init__.py index f0a59c8..4c3c2d5 100644 --- a/torch_mimicry/__init__.py +++ b/torch_mimicry/__init__.py @@ -1,3 +1,3 @@ from torch_mimicry import nets, training, metrics, datasets, modules -__version__ = "0.1.14" +__version__ = "0.1.16" diff --git a/torch_mimicry/nets/sagan/__init__.py b/torch_mimicry/nets/sagan/__init__.py new file mode 100644 index 0000000..20fecf1 --- /dev/null +++ b/torch_mimicry/nets/sagan/__init__.py @@ -0,0 +1,3 @@ +from .sagan_128 import * +from .sagan_32 import * +from .sagan_base import *