gitextract_gjrwr8nv/

├── .gitignore
├── LICENSE
├── MANIFEST.in
├── readme.md
├── setup.cfg
├── setup.py
└── torchtools/
    ├── __init__.py
    ├── lr_scheduler/
    │   ├── __init__.py
    │   ├── delayed.py
    │   └── inverse_sqrt.py
    ├── nn/
    │   ├── __init__.py
    │   ├── adain.py
    │   ├── alias_free_activation.py
    │   ├── equal_layers.py
    │   ├── evonorm2d.py
    │   ├── fourier_features.py
    │   ├── functional/
    │   │   ├── __init__.py
    │   │   ├── gradient_penalty.py
    │   │   ├── magnitude_preserving.py
    │   │   ├── perceptual.py
    │   │   └── vq.py
    │   ├── gp_loss.py
    │   ├── haar_dwt.py
    │   ├── magnitude_preserving.py
    │   ├── mish.py
    │   ├── modulation.py
    │   ├── perceptual.py
    │   ├── pixel_normalzation.py
    │   ├── pos_embeddings.py
    │   ├── simple_self_attention.py
    │   ├── stylegan2/
    │   │   ├── __init__.py
    │   │   ├── upfirdn2d.cpp
    │   │   ├── upfirdn2d.py
    │   │   └── upfirdn2d_kernel.cu
    │   ├── transformers.py
    │   └── vq.py
    ├── optim/
    │   ├── __init__.py
    │   ├── lamb.py
    │   ├── lookahead.py
    │   ├── novograd.py
    │   ├── over9000.py
    │   ├── radam.py
    │   ├── ralamb.py
    │   └── ranger.py
    ├── transforms/
    │   ├── __init__.py
    │   ├── models/
    │   │   ├── __init__.py
    │   │   └── saliency_model_v9.pt
    │   └── smart_crop.py
    └── utils/
        ├── __init__.py
        ├── diffusion.py
        ├── diffusion2.py
        ├── gamma_parametrization.py
        └── weight_normalization.py