Skip to content
Snippets Groups Projects
Commit 1ac5e8fc authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[init] corrected the name to import both trainer and architectures

parent a7abd4e9
No related branches found
No related tags found
No related merge requests found
...@@ -4,9 +4,6 @@ from .DCGAN import DCGAN_discriminator ...@@ -4,9 +4,6 @@ from .DCGAN import DCGAN_discriminator
from .ConditionalGAN import ConditionalGAN_generator from .ConditionalGAN import ConditionalGAN_generator
from .ConditionalGAN import ConditionalGAN_discriminator from .ConditionalGAN import ConditionalGAN_discriminator
from .ConditionalGAN2 import ConditionalGAN_generator2
from .ConditionalGAN2 import ConditionalGAN_discriminator2
from .DCGAN import weights_init from .DCGAN import weights_init
# gets sphinx autodoc done right - don't remove it # gets sphinx autodoc done right - don't remove it
......
from .DCGANTrainer import DCGANTrainer from .DCGANTrainer import DCGANTrainer
from .ConditionalGANTrainer import ConditionalGANTrainer from .ConditionalGANTrainer import ConditionalGANTrainer
from .ConditionalGANTrainer2 import ConditionalGANTrainer2
# gets sphinx autodoc done right - don't remove it # gets sphinx autodoc done right - don't remove it
__all__ = [_ for _ in dir() if not _.startswith('_')] __all__ = [_ for _ in dir() if not _.startswith('_')]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment