pkg src/ (#1010)

Lightning-Universe · May 18, 2023 · a0bad20 · a0bad20
1 parent 0a6955e
commit a0bad20
Show file tree

Hide file tree

Showing 177 changed files with 17 additions and 19 deletions.
diff --git a/.azure/gpu-tests.yml b/.azure/gpu-tests.yml
@@ -12,23 +12,26 @@ pr:
   - release/*
 
 jobs:
-  - job: pytest
+  - job: testing
     # how long to run the job before automatically cancelling
     timeoutInMinutes: "45"
     # how much time to give 'run always even if cancelled tasks' before stopping them
     cancelTimeoutInMinutes: "2"
 
     pool: lit-rtx-3090
     variables:
-      DEVICES: $( python -c 'print("$(Agent.Name)".split("_")[-1])' )
+      DEVICES: $( python -c 'name = "$(Agent.Name)" ; gpus = name.split("_")[-1] if "_" in name else "0,1"; print(gpus)' )
       DEBIAN_FRONTEND: "noninteractive"
       TZ: "Europe/Amsterdam"
+      TORCH_HOME: "/var/tmp/torch"
+      TRANSFORMERS_CACHE: "/var/tmp/huggingface"
+      FREEZE_REQUIREMENTS: 1
 
     container:
       #image: "pytorchlightning/pytorch_lightning:base-cuda-py3.9-torch1.12-cuda11.6.1"
       image: "pytorch/pytorch:1.12.1-cuda11.3-cudnn8-runtime"
       #endpoint: azureContainerRegistryConnection
-      options: "--gpus=all --shm-size=16g -v /usr/bin/docker:/tmp/docker:ro"
+      options: "--gpus=all --shm-size=16g -v /usr/bin/docker:/tmp/docker:ro -v /var/tmp:/var/tmp"
 
     workspace:
       clean: all

diff --git a/.github/workflows/ci-tests.yml b/.github/workflows/ci-tests.yml
@@ -26,7 +26,8 @@ jobs:
         # todo: seems there are some issues with py3.10 while install dependencies
         python-version: ["3.7", "3.9"]
         requires: ['oldest', 'latest']
-
+    env:
+      FREEZE_REQUIREMENTS: 1
     # Timeout: https://stackoverflow.com/a/59076067/4521646
     # the reason for high number is MUCH slower tests on macOS and py3.8
     timeout-minutes: 50
@@ -69,10 +70,11 @@ jobs:
         pip list
 
     - name: Test Package [only]
+      working-directory: ./src
       run: |
         pip install pytest -q
         # TODO: package shall be fine to run full without any ignores
-        python -m pytest pl_bolts \
+        python -m pytest . \
           --ignore=pl_bolts/datamodules \
           --ignore=pl_bolts/datasets \
           --ignore=pl_bolts/models/self_supervised/amdim/transforms.py \
@@ -86,7 +88,7 @@ jobs:
       uses: actions/cache@v3
       with:
         path: ${{ steps.pip-cache.outputs.dir }}
-        key: ${{ runner.os }}-pip-py${{ matrix.python-version }}-${{ matrix.requires }}-${{ hashFiles('requirements.txt') }}-${{ hashFiles('requirements/modules.txt') }}
+        key: ${{ runner.os }}-pip-py${{ matrix.python-version }}-${{ matrix.requires }}-${{ hashFiles('requirements.txt') }}-${{ hashFiles('requirements/*.txt') }}
         restore-keys: |
           ${{ runner.os }}-pip-py${{ matrix.python-version }}-${{ matrix.requires }}-
 
@@ -113,16 +115,7 @@ jobs:
 
     - name: Tests
       run: |
-        python -m pytest tests -v --cov=pl_bolts \
-          --junitxml=junit/test-results-${{ runner.os }}-${{ matrix.python-version }}-${{ matrix.requires }}.xml
-
-    - name: Upload pytest test results
-      uses: actions/upload-artifact@v3
-      with:
-        name: pytest-results-${{ runner.os }}-${{ matrix.python-version }}-${{ matrix.requires }}
-        path: junit/test-results-${{ runner.os }}-${{ matrix.python-version }}-${{ matrix.requires }}.xml
-      # Use always() to always run this step to publish test results when there are test failures
-      if: always()
+        python -m pytest tests -v --cov=pl_bolts
 
     - name: Statistics
       if: success()

diff --git a/pyproject.toml b/pyproject.toml
@@ -50,7 +50,7 @@ ignore-init-module-imports = true
 unfixable = ["F401"]
 
 [tool.mypy]
-files = ["pl_bolts"]
+files = ["src/pl_bolts"]
 disallow_untyped_defs = "True"
 ignore_missing_imports = "True"
 install_types = "True"

diff --git a/setup.py b/setup.py
@@ -12,7 +12,7 @@
 
 
 def _load_py_module(fname, pkg="pl_bolts"):
-    spec = spec_from_file_location(os.path.join(pkg, fname), os.path.join(_PATH_ROOT, pkg, fname))
+    spec = spec_from_file_location(os.path.join(pkg, fname), os.path.join(_PATH_ROOT, "src", pkg, fname))
     py = module_from_spec(spec)
     spec.loader.exec_module(py)
     return py
@@ -115,6 +115,7 @@ def _prepare_extras():
         "loggers": _load_requirements(path_dir=_PATH_REQUIRE, file_name="loggers.txt"),
         "models": _load_requirements(path_dir=_PATH_REQUIRE, file_name="models.txt"),
         "test": _load_requirements(path_dir=_PATH_REQUIRE, file_name="test.txt"),
+        "typing": _load_requirements(path_dir=_PATH_REQUIRE, file_name="typing.txt"),
     }
     extras["extra"] = extras["models"] + extras["loggers"]
     extras["dev"] = extras["extra"] + extras["test"]
@@ -142,7 +143,8 @@ def _prepare_extras():
     url=about.__homepage__,
     download_url="https://github.com/PyTorchLightning/lightning-bolts",
     license=about.__license__,
-    packages=find_packages(exclude=["tests", "docs"]),
+    package_dir={"": "src"},
+    packages=find_packages(where="src"),
     long_description=long_description,
     long_description_content_type="text/markdown",
     include_package_data=True,

diff --git a/pl_bolts/__about__.py → src/pl_bolts/__about__.py b/pl_bolts/__about__.py → src/pl_bolts/__about__.py
diff --git a/pl_bolts/__init__.py → src/pl_bolts/__init__.py b/pl_bolts/__init__.py → src/pl_bolts/__init__.py
diff --git a/pl_bolts/callbacks/__init__.py → src/pl_bolts/callbacks/__init__.py b/pl_bolts/callbacks/__init__.py → src/pl_bolts/callbacks/__init__.py
diff --git a/pl_bolts/callbacks/byol_updates.py → src/pl_bolts/callbacks/byol_updates.py b/pl_bolts/callbacks/byol_updates.py → src/pl_bolts/callbacks/byol_updates.py
diff --git a/pl_bolts/callbacks/data_monitor.py → src/pl_bolts/callbacks/data_monitor.py b/pl_bolts/callbacks/data_monitor.py → src/pl_bolts/callbacks/data_monitor.py
diff --git a/pl_bolts/callbacks/knn_online.py → src/pl_bolts/callbacks/knn_online.py b/pl_bolts/callbacks/knn_online.py → src/pl_bolts/callbacks/knn_online.py
diff --git a/pl_bolts/callbacks/printing.py → src/pl_bolts/callbacks/printing.py b/pl_bolts/callbacks/printing.py → src/pl_bolts/callbacks/printing.py
diff --git a/pl_bolts/callbacks/sparseml.py → src/pl_bolts/callbacks/sparseml.py b/pl_bolts/callbacks/sparseml.py → src/pl_bolts/callbacks/sparseml.py
diff --git a/pl_bolts/callbacks/ssl_online.py → src/pl_bolts/callbacks/ssl_online.py b/pl_bolts/callbacks/ssl_online.py → src/pl_bolts/callbacks/ssl_online.py
diff --git a/pl_bolts/callbacks/torch_ort.py → src/pl_bolts/callbacks/torch_ort.py b/pl_bolts/callbacks/torch_ort.py → src/pl_bolts/callbacks/torch_ort.py
diff --git a/pl_bolts/callbacks/variational.py → src/pl_bolts/callbacks/variational.py b/pl_bolts/callbacks/variational.py → src/pl_bolts/callbacks/variational.py
diff --git a/pl_bolts/callbacks/verification/__init__.py → ..._bolts/callbacks/verification/__init__.py b/pl_bolts/callbacks/verification/__init__.py → ..._bolts/callbacks/verification/__init__.py
diff --git a/pl_bolts/callbacks/verification/base.py → src/pl_bolts/callbacks/verification/base.py b/pl_bolts/callbacks/verification/base.py → src/pl_bolts/callbacks/verification/base.py
diff --git a/.../callbacks/verification/batch_gradient.py → .../callbacks/verification/batch_gradient.py b/.../callbacks/verification/batch_gradient.py → .../callbacks/verification/batch_gradient.py
diff --git a/pl_bolts/callbacks/vision/__init__.py → src/pl_bolts/callbacks/vision/__init__.py b/pl_bolts/callbacks/vision/__init__.py → src/pl_bolts/callbacks/vision/__init__.py
diff --git a/pl_bolts/callbacks/vision/confused_logit.py → ..._bolts/callbacks/vision/confused_logit.py b/pl_bolts/callbacks/vision/confused_logit.py → ..._bolts/callbacks/vision/confused_logit.py
diff --git a/...olts/callbacks/vision/image_generation.py → ...olts/callbacks/vision/image_generation.py b/...olts/callbacks/vision/image_generation.py → ...olts/callbacks/vision/image_generation.py
diff --git a/pl_bolts/callbacks/vision/sr_image_logger.py → ...bolts/callbacks/vision/sr_image_logger.py b/pl_bolts/callbacks/vision/sr_image_logger.py → ...bolts/callbacks/vision/sr_image_logger.py
diff --git a/pl_bolts/datamodules/__init__.py → src/pl_bolts/datamodules/__init__.py b/pl_bolts/datamodules/__init__.py → src/pl_bolts/datamodules/__init__.py
diff --git a/pl_bolts/datamodules/async_dataloader.py → src/pl_bolts/datamodules/async_dataloader.py b/pl_bolts/datamodules/async_dataloader.py → src/pl_bolts/datamodules/async_dataloader.py
diff --git a/...s/datamodules/binary_emnist_datamodule.py → ...s/datamodules/binary_emnist_datamodule.py b/...s/datamodules/binary_emnist_datamodule.py → ...s/datamodules/binary_emnist_datamodule.py
diff --git a/...ts/datamodules/binary_mnist_datamodule.py → ...ts/datamodules/binary_mnist_datamodule.py b/...ts/datamodules/binary_mnist_datamodule.py → ...ts/datamodules/binary_mnist_datamodule.py
diff --git a/pl_bolts/datamodules/cifar10_datamodule.py → ...l_bolts/datamodules/cifar10_datamodule.py b/pl_bolts/datamodules/cifar10_datamodule.py → ...l_bolts/datamodules/cifar10_datamodule.py
diff --git a/...olts/datamodules/cityscapes_datamodule.py → ...olts/datamodules/cityscapes_datamodule.py b/...olts/datamodules/cityscapes_datamodule.py → ...olts/datamodules/cityscapes_datamodule.py
diff --git a/pl_bolts/datamodules/emnist_datamodule.py → ...pl_bolts/datamodules/emnist_datamodule.py b/pl_bolts/datamodules/emnist_datamodule.py → ...pl_bolts/datamodules/emnist_datamodule.py
diff --git a/pl_bolts/datamodules/experience_source.py → ...pl_bolts/datamodules/experience_source.py b/pl_bolts/datamodules/experience_source.py → ...pl_bolts/datamodules/experience_source.py
diff --git a/...s/datamodules/fashion_mnist_datamodule.py → ...s/datamodules/fashion_mnist_datamodule.py b/...s/datamodules/fashion_mnist_datamodule.py → ...s/datamodules/fashion_mnist_datamodule.py
diff --git a/pl_bolts/datamodules/imagenet_datamodule.py → ..._bolts/datamodules/imagenet_datamodule.py b/pl_bolts/datamodules/imagenet_datamodule.py → ..._bolts/datamodules/imagenet_datamodule.py
diff --git a/pl_bolts/datamodules/kitti_datamodule.py → src/pl_bolts/datamodules/kitti_datamodule.py b/pl_bolts/datamodules/kitti_datamodule.py → src/pl_bolts/datamodules/kitti_datamodule.py
diff --git a/pl_bolts/datamodules/mnist_datamodule.py → src/pl_bolts/datamodules/mnist_datamodule.py b/pl_bolts/datamodules/mnist_datamodule.py → src/pl_bolts/datamodules/mnist_datamodule.py
diff --git a/pl_bolts/datamodules/sklearn_datamodule.py → ...l_bolts/datamodules/sklearn_datamodule.py b/pl_bolts/datamodules/sklearn_datamodule.py → ...l_bolts/datamodules/sklearn_datamodule.py
diff --git a/pl_bolts/datamodules/sr_datamodule.py → src/pl_bolts/datamodules/sr_datamodule.py b/pl_bolts/datamodules/sr_datamodule.py → src/pl_bolts/datamodules/sr_datamodule.py
diff --git a/...ts/datamodules/ssl_imagenet_datamodule.py → ...ts/datamodules/ssl_imagenet_datamodule.py b/...ts/datamodules/ssl_imagenet_datamodule.py → ...ts/datamodules/ssl_imagenet_datamodule.py
diff --git a/pl_bolts/datamodules/stl10_datamodule.py → src/pl_bolts/datamodules/stl10_datamodule.py b/pl_bolts/datamodules/stl10_datamodule.py → src/pl_bolts/datamodules/stl10_datamodule.py
diff --git a/pl_bolts/datamodules/vision_datamodule.py → ...pl_bolts/datamodules/vision_datamodule.py b/pl_bolts/datamodules/vision_datamodule.py → ...pl_bolts/datamodules/vision_datamodule.py
diff --git a/...ts/datamodules/vocdetection_datamodule.py → ...ts/datamodules/vocdetection_datamodule.py b/...ts/datamodules/vocdetection_datamodule.py → ...ts/datamodules/vocdetection_datamodule.py
diff --git a/pl_bolts/datasets/__init__.py → src/pl_bolts/datasets/__init__.py b/pl_bolts/datasets/__init__.py → src/pl_bolts/datasets/__init__.py
diff --git a/pl_bolts/datasets/array_dataset.py → src/pl_bolts/datasets/array_dataset.py b/pl_bolts/datasets/array_dataset.py → src/pl_bolts/datasets/array_dataset.py
diff --git a/pl_bolts/datasets/base_dataset.py → src/pl_bolts/datasets/base_dataset.py b/pl_bolts/datasets/base_dataset.py → src/pl_bolts/datasets/base_dataset.py
diff --git a/pl_bolts/datasets/cifar10_dataset.py → src/pl_bolts/datasets/cifar10_dataset.py b/pl_bolts/datasets/cifar10_dataset.py → src/pl_bolts/datasets/cifar10_dataset.py
diff --git a/pl_bolts/datasets/concat_dataset.py → src/pl_bolts/datasets/concat_dataset.py b/pl_bolts/datasets/concat_dataset.py → src/pl_bolts/datasets/concat_dataset.py
diff --git a/pl_bolts/datasets/dummy_dataset.py → src/pl_bolts/datasets/dummy_dataset.py b/pl_bolts/datasets/dummy_dataset.py → src/pl_bolts/datasets/dummy_dataset.py
diff --git a/pl_bolts/datasets/emnist_dataset.py → src/pl_bolts/datasets/emnist_dataset.py b/pl_bolts/datasets/emnist_dataset.py → src/pl_bolts/datasets/emnist_dataset.py
diff --git a/pl_bolts/datasets/imagenet_dataset.py → src/pl_bolts/datasets/imagenet_dataset.py b/pl_bolts/datasets/imagenet_dataset.py → src/pl_bolts/datasets/imagenet_dataset.py
diff --git a/pl_bolts/datasets/kitti_dataset.py → src/pl_bolts/datasets/kitti_dataset.py b/pl_bolts/datasets/kitti_dataset.py → src/pl_bolts/datasets/kitti_dataset.py
diff --git a/pl_bolts/datasets/mnist_dataset.py → src/pl_bolts/datasets/mnist_dataset.py b/pl_bolts/datasets/mnist_dataset.py → src/pl_bolts/datasets/mnist_dataset.py
diff --git a/pl_bolts/datasets/sr_celeba_dataset.py → src/pl_bolts/datasets/sr_celeba_dataset.py b/pl_bolts/datasets/sr_celeba_dataset.py → src/pl_bolts/datasets/sr_celeba_dataset.py
diff --git a/pl_bolts/datasets/sr_dataset_mixin.py → src/pl_bolts/datasets/sr_dataset_mixin.py b/pl_bolts/datasets/sr_dataset_mixin.py → src/pl_bolts/datasets/sr_dataset_mixin.py
diff --git a/pl_bolts/datasets/sr_mnist_dataset.py → src/pl_bolts/datasets/sr_mnist_dataset.py b/pl_bolts/datasets/sr_mnist_dataset.py → src/pl_bolts/datasets/sr_mnist_dataset.py
diff --git a/pl_bolts/datasets/sr_stl10_dataset.py → src/pl_bolts/datasets/sr_stl10_dataset.py b/pl_bolts/datasets/sr_stl10_dataset.py → src/pl_bolts/datasets/sr_stl10_dataset.py
diff --git a/pl_bolts/datasets/ssl_amdim_datasets.py → src/pl_bolts/datasets/ssl_amdim_datasets.py b/pl_bolts/datasets/ssl_amdim_datasets.py → src/pl_bolts/datasets/ssl_amdim_datasets.py
diff --git a/pl_bolts/datasets/utils.py → src/pl_bolts/datasets/utils.py b/pl_bolts/datasets/utils.py → src/pl_bolts/datasets/utils.py
diff --git a/pl_bolts/losses/__init__.py → src/pl_bolts/losses/__init__.py b/pl_bolts/losses/__init__.py → src/pl_bolts/losses/__init__.py
diff --git a/pl_bolts/losses/object_detection.py → src/pl_bolts/losses/object_detection.py b/pl_bolts/losses/object_detection.py → src/pl_bolts/losses/object_detection.py
diff --git a/pl_bolts/losses/rl.py → src/pl_bolts/losses/rl.py b/pl_bolts/losses/rl.py → src/pl_bolts/losses/rl.py
diff --git a/pl_bolts/losses/self_supervised_learning.py → ..._bolts/losses/self_supervised_learning.py b/pl_bolts/losses/self_supervised_learning.py → ..._bolts/losses/self_supervised_learning.py
diff --git a/pl_bolts/metrics/__init__.py → src/pl_bolts/metrics/__init__.py b/pl_bolts/metrics/__init__.py → src/pl_bolts/metrics/__init__.py
diff --git a/pl_bolts/metrics/aggregation.py → src/pl_bolts/metrics/aggregation.py b/pl_bolts/metrics/aggregation.py → src/pl_bolts/metrics/aggregation.py
diff --git a/pl_bolts/metrics/object_detection.py → src/pl_bolts/metrics/object_detection.py b/pl_bolts/metrics/object_detection.py → src/pl_bolts/metrics/object_detection.py
diff --git a/pl_bolts/models/__init__.py → src/pl_bolts/models/__init__.py b/pl_bolts/models/__init__.py → src/pl_bolts/models/__init__.py
diff --git a/pl_bolts/models/autoencoders/__init__.py → src/pl_bolts/models/autoencoders/__init__.py b/pl_bolts/models/autoencoders/__init__.py → src/pl_bolts/models/autoencoders/__init__.py
diff --git a/.../models/autoencoders/basic_ae/__init__.py → .../models/autoencoders/basic_ae/__init__.py b/.../models/autoencoders/basic_ae/__init__.py → .../models/autoencoders/basic_ae/__init__.py
diff --git a/.../autoencoders/basic_ae/basic_ae_module.py → .../autoencoders/basic_ae/basic_ae_module.py b/.../autoencoders/basic_ae/basic_ae_module.py → .../autoencoders/basic_ae/basic_ae_module.py
diff --git a/...models/autoencoders/basic_vae/__init__.py → ...models/autoencoders/basic_vae/__init__.py b/...models/autoencoders/basic_vae/__init__.py → ...models/autoencoders/basic_vae/__init__.py
diff --git a/...utoencoders/basic_vae/basic_vae_module.py → ...utoencoders/basic_vae/basic_vae_module.py b/...utoencoders/basic_vae/basic_vae_module.py → ...utoencoders/basic_vae/basic_vae_module.py
diff --git a/pl_bolts/models/autoencoders/components.py → ...l_bolts/models/autoencoders/components.py b/pl_bolts/models/autoencoders/components.py → ...l_bolts/models/autoencoders/components.py
diff --git a/pl_bolts/models/detection/__init__.py → src/pl_bolts/models/detection/__init__.py b/pl_bolts/models/detection/__init__.py → src/pl_bolts/models/detection/__init__.py
diff --git a/...s/models/detection/components/__init__.py → ...s/models/detection/components/__init__.py b/...s/models/detection/components/__init__.py → ...s/models/detection/components/__init__.py
diff --git a/...detection/components/_supported_models.py → ...detection/components/_supported_models.py b/...detection/components/_supported_models.py → ...detection/components/_supported_models.py
diff --git a/...ction/components/torchvision_backbones.py → ...ction/components/torchvision_backbones.py b/...ction/components/torchvision_backbones.py → ...ction/components/torchvision_backbones.py
diff --git a/.../models/detection/faster_rcnn/__init__.py → .../models/detection/faster_rcnn/__init__.py b/.../models/detection/faster_rcnn/__init__.py → .../models/detection/faster_rcnn/__init__.py
diff --git a/...models/detection/faster_rcnn/backbones.py → ...models/detection/faster_rcnn/backbones.py b/...models/detection/faster_rcnn/backbones.py → ...models/detection/faster_rcnn/backbones.py
diff --git a/...tection/faster_rcnn/faster_rcnn_module.py → ...tection/faster_rcnn/faster_rcnn_module.py b/...tection/faster_rcnn/faster_rcnn_module.py → ...tection/faster_rcnn/faster_rcnn_module.py
diff --git a/...ts/models/detection/retinanet/__init__.py → ...ts/models/detection/retinanet/__init__.py b/...ts/models/detection/retinanet/__init__.py → ...ts/models/detection/retinanet/__init__.py
diff --git a/...s/models/detection/retinanet/backbones.py → ...s/models/detection/retinanet/backbones.py b/...s/models/detection/retinanet/backbones.py → ...s/models/detection/retinanet/backbones.py
diff --git a/...s/detection/retinanet/retinanet_module.py → ...s/detection/retinanet/retinanet_module.py b/...s/detection/retinanet/retinanet_module.py → ...s/detection/retinanet/retinanet_module.py
diff --git a/pl_bolts/models/detection/yolo/__init__.py → ...l_bolts/models/detection/yolo/__init__.py b/pl_bolts/models/detection/yolo/__init__.py → ...l_bolts/models/detection/yolo/__init__.py
diff --git a/...olts/models/detection/yolo/yolo_config.py → ...olts/models/detection/yolo/yolo_config.py b/...olts/models/detection/yolo/yolo_config.py → ...olts/models/detection/yolo/yolo_config.py
diff --git a/...olts/models/detection/yolo/yolo_layers.py → ...olts/models/detection/yolo/yolo_layers.py b/...olts/models/detection/yolo/yolo_layers.py → ...olts/models/detection/yolo/yolo_layers.py
diff --git a/...olts/models/detection/yolo/yolo_module.py → ...olts/models/detection/yolo/yolo_module.py b/...olts/models/detection/yolo/yolo_module.py → ...olts/models/detection/yolo/yolo_module.py
diff --git a/pl_bolts/models/gans/__init__.py → src/pl_bolts/models/gans/__init__.py b/pl_bolts/models/gans/__init__.py → src/pl_bolts/models/gans/__init__.py
diff --git a/pl_bolts/models/gans/basic/__init__.py → src/pl_bolts/models/gans/basic/__init__.py b/pl_bolts/models/gans/basic/__init__.py → src/pl_bolts/models/gans/basic/__init__.py
diff --git a/...lts/models/gans/basic/basic_gan_module.py → ...lts/models/gans/basic/basic_gan_module.py b/...lts/models/gans/basic/basic_gan_module.py → ...lts/models/gans/basic/basic_gan_module.py
diff --git a/pl_bolts/models/gans/basic/components.py → src/pl_bolts/models/gans/basic/components.py b/pl_bolts/models/gans/basic/components.py → src/pl_bolts/models/gans/basic/components.py
diff --git a/pl_bolts/models/gans/dcgan/__init__.py → src/pl_bolts/models/gans/dcgan/__init__.py b/pl_bolts/models/gans/dcgan/__init__.py → src/pl_bolts/models/gans/dcgan/__init__.py
diff --git a/pl_bolts/models/gans/dcgan/components.py → src/pl_bolts/models/gans/dcgan/components.py b/pl_bolts/models/gans/dcgan/components.py → src/pl_bolts/models/gans/dcgan/components.py
diff --git a/pl_bolts/models/gans/dcgan/dcgan_module.py → ...l_bolts/models/gans/dcgan/dcgan_module.py b/pl_bolts/models/gans/dcgan/dcgan_module.py → ...l_bolts/models/gans/dcgan/dcgan_module.py
diff --git a/pl_bolts/models/gans/pix2pix/__init__.py → src/pl_bolts/models/gans/pix2pix/__init__.py b/pl_bolts/models/gans/pix2pix/__init__.py → src/pl_bolts/models/gans/pix2pix/__init__.py
diff --git a/pl_bolts/models/gans/pix2pix/components.py → ...l_bolts/models/gans/pix2pix/components.py b/pl_bolts/models/gans/pix2pix/components.py → ...l_bolts/models/gans/pix2pix/components.py
diff --git a/...lts/models/gans/pix2pix/pix2pix_module.py → ...lts/models/gans/pix2pix/pix2pix_module.py b/...lts/models/gans/pix2pix/pix2pix_module.py → ...lts/models/gans/pix2pix/pix2pix_module.py
diff --git a/pl_bolts/models/gans/srgan/__init__.py → src/pl_bolts/models/gans/srgan/__init__.py b/pl_bolts/models/gans/srgan/__init__.py → src/pl_bolts/models/gans/srgan/__init__.py
diff --git a/pl_bolts/models/gans/srgan/components.py → src/pl_bolts/models/gans/srgan/components.py b/pl_bolts/models/gans/srgan/components.py → src/pl_bolts/models/gans/srgan/components.py
diff --git a/pl_bolts/models/gans/srgan/srgan_module.py → ...l_bolts/models/gans/srgan/srgan_module.py b/pl_bolts/models/gans/srgan/srgan_module.py → ...l_bolts/models/gans/srgan/srgan_module.py
diff --git a/...olts/models/gans/srgan/srresnet_module.py → ...olts/models/gans/srgan/srresnet_module.py b/...olts/models/gans/srgan/srresnet_module.py → ...olts/models/gans/srgan/srresnet_module.py
diff --git a/pl_bolts/models/mnist_module.py → src/pl_bolts/models/mnist_module.py b/pl_bolts/models/mnist_module.py → src/pl_bolts/models/mnist_module.py
diff --git a/pl_bolts/models/regression/__init__.py → src/pl_bolts/models/regression/__init__.py b/pl_bolts/models/regression/__init__.py → src/pl_bolts/models/regression/__init__.py
diff --git a/...ts/models/regression/linear_regression.py → ...ts/models/regression/linear_regression.py b/...ts/models/regression/linear_regression.py → ...ts/models/regression/linear_regression.py
diff --git a/.../models/regression/logistic_regression.py → .../models/regression/logistic_regression.py b/.../models/regression/logistic_regression.py → .../models/regression/logistic_regression.py
diff --git a/pl_bolts/models/rl/__init__.py → src/pl_bolts/models/rl/__init__.py b/pl_bolts/models/rl/__init__.py → src/pl_bolts/models/rl/__init__.py
diff --git a/...models/rl/advantage_actor_critic_model.py → ...models/rl/advantage_actor_critic_model.py b/...models/rl/advantage_actor_critic_model.py → ...models/rl/advantage_actor_critic_model.py
diff --git a/pl_bolts/models/rl/common/__init__.py → src/pl_bolts/models/rl/common/__init__.py b/pl_bolts/models/rl/common/__init__.py → src/pl_bolts/models/rl/common/__init__.py
diff --git a/pl_bolts/models/rl/common/agents.py → src/pl_bolts/models/rl/common/agents.py b/pl_bolts/models/rl/common/agents.py → src/pl_bolts/models/rl/common/agents.py
diff --git a/pl_bolts/models/rl/common/cli.py → src/pl_bolts/models/rl/common/cli.py b/pl_bolts/models/rl/common/cli.py → src/pl_bolts/models/rl/common/cli.py
diff --git a/pl_bolts/models/rl/common/distributions.py → ...l_bolts/models/rl/common/distributions.py b/pl_bolts/models/rl/common/distributions.py → ...l_bolts/models/rl/common/distributions.py
diff --git a/pl_bolts/models/rl/common/gym_wrappers.py → ...pl_bolts/models/rl/common/gym_wrappers.py b/pl_bolts/models/rl/common/gym_wrappers.py → ...pl_bolts/models/rl/common/gym_wrappers.py
diff --git a/pl_bolts/models/rl/common/memory.py → src/pl_bolts/models/rl/common/memory.py b/pl_bolts/models/rl/common/memory.py → src/pl_bolts/models/rl/common/memory.py
diff --git a/pl_bolts/models/rl/common/networks.py → src/pl_bolts/models/rl/common/networks.py b/pl_bolts/models/rl/common/networks.py → src/pl_bolts/models/rl/common/networks.py
diff --git a/pl_bolts/models/rl/double_dqn_model.py → src/pl_bolts/models/rl/double_dqn_model.py b/pl_bolts/models/rl/double_dqn_model.py → src/pl_bolts/models/rl/double_dqn_model.py
diff --git a/pl_bolts/models/rl/dqn_model.py → src/pl_bolts/models/rl/dqn_model.py b/pl_bolts/models/rl/dqn_model.py → src/pl_bolts/models/rl/dqn_model.py
diff --git a/pl_bolts/models/rl/dueling_dqn_model.py → src/pl_bolts/models/rl/dueling_dqn_model.py b/pl_bolts/models/rl/dueling_dqn_model.py → src/pl_bolts/models/rl/dueling_dqn_model.py
diff --git a/pl_bolts/models/rl/noisy_dqn_model.py → src/pl_bolts/models/rl/noisy_dqn_model.py b/pl_bolts/models/rl/noisy_dqn_model.py → src/pl_bolts/models/rl/noisy_dqn_model.py
diff --git a/pl_bolts/models/rl/per_dqn_model.py → src/pl_bolts/models/rl/per_dqn_model.py b/pl_bolts/models/rl/per_dqn_model.py → src/pl_bolts/models/rl/per_dqn_model.py
diff --git a/pl_bolts/models/rl/ppo_model.py → src/pl_bolts/models/rl/ppo_model.py b/pl_bolts/models/rl/ppo_model.py → src/pl_bolts/models/rl/ppo_model.py
diff --git a/pl_bolts/models/rl/reinforce_model.py → src/pl_bolts/models/rl/reinforce_model.py b/pl_bolts/models/rl/reinforce_model.py → src/pl_bolts/models/rl/reinforce_model.py
diff --git a/pl_bolts/models/rl/sac_model.py → src/pl_bolts/models/rl/sac_model.py b/pl_bolts/models/rl/sac_model.py → src/pl_bolts/models/rl/sac_model.py
diff --git a/...odels/rl/vanilla_policy_gradient_model.py → ...odels/rl/vanilla_policy_gradient_model.py b/...odels/rl/vanilla_policy_gradient_model.py → ...odels/rl/vanilla_policy_gradient_model.py
diff --git a/pl_bolts/models/self_supervised/__init__.py → ..._bolts/models/self_supervised/__init__.py b/pl_bolts/models/self_supervised/__init__.py → ..._bolts/models/self_supervised/__init__.py
diff --git a/.../models/self_supervised/amdim/__init__.py → .../models/self_supervised/amdim/__init__.py b/.../models/self_supervised/amdim/__init__.py → .../models/self_supervised/amdim/__init__.py
diff --git a/...els/self_supervised/amdim/amdim_module.py → ...els/self_supervised/amdim/amdim_module.py b/...els/self_supervised/amdim/amdim_module.py → ...els/self_supervised/amdim/amdim_module.py
diff --git a/.../models/self_supervised/amdim/datasets.py → .../models/self_supervised/amdim/datasets.py b/.../models/self_supervised/amdim/datasets.py → .../models/self_supervised/amdim/datasets.py
diff --git a/.../models/self_supervised/amdim/networks.py → .../models/self_supervised/amdim/networks.py b/.../models/self_supervised/amdim/networks.py → .../models/self_supervised/amdim/networks.py
diff --git a/...odels/self_supervised/amdim/transforms.py → ...odels/self_supervised/amdim/transforms.py b/...odels/self_supervised/amdim/transforms.py → ...odels/self_supervised/amdim/transforms.py
diff --git a/...s/models/self_supervised/byol/__init__.py → ...s/models/self_supervised/byol/__init__.py b/...s/models/self_supervised/byol/__init__.py → ...s/models/self_supervised/byol/__init__.py
diff --git a/...odels/self_supervised/byol/byol_module.py → ...odels/self_supervised/byol/byol_module.py b/...odels/self_supervised/byol/byol_module.py → ...odels/self_supervised/byol/byol_module.py
diff --git a/...lts/models/self_supervised/byol/models.py → ...lts/models/self_supervised/byol/models.py b/...lts/models/self_supervised/byol/models.py → ...lts/models/self_supervised/byol/models.py
diff --git a/...ts/models/self_supervised/cpc/__init__.py → ...ts/models/self_supervised/cpc/__init__.py b/...ts/models/self_supervised/cpc/__init__.py → ...ts/models/self_supervised/cpc/__init__.py
diff --git a/...dels/self_supervised/cpc/cpc_finetuner.py → ...dels/self_supervised/cpc/cpc_finetuner.py b/...dels/self_supervised/cpc/cpc_finetuner.py → ...dels/self_supervised/cpc/cpc_finetuner.py
diff --git a/.../models/self_supervised/cpc/cpc_module.py → .../models/self_supervised/cpc/cpc_module.py b/.../models/self_supervised/cpc/cpc_module.py → .../models/self_supervised/cpc/cpc_module.py
diff --git a/...ts/models/self_supervised/cpc/networks.py → ...ts/models/self_supervised/cpc/networks.py b/...ts/models/self_supervised/cpc/networks.py → ...ts/models/self_supervised/cpc/networks.py
diff --git a/.../models/self_supervised/cpc/transforms.py → .../models/self_supervised/cpc/transforms.py b/.../models/self_supervised/cpc/transforms.py → .../models/self_supervised/cpc/transforms.py
diff --git a/pl_bolts/models/self_supervised/evaluator.py → ...bolts/models/self_supervised/evaluator.py b/pl_bolts/models/self_supervised/evaluator.py → ...bolts/models/self_supervised/evaluator.py
diff --git a/pl_bolts/models/self_supervised/moco/LICENSE → ...bolts/models/self_supervised/moco/LICENSE b/pl_bolts/models/self_supervised/moco/LICENSE → ...bolts/models/self_supervised/moco/LICENSE
diff --git a/...s/models/self_supervised/moco/__init__.py → ...s/models/self_supervised/moco/__init__.py b/...s/models/self_supervised/moco/__init__.py → ...s/models/self_supervised/moco/__init__.py
diff --git a/.../models/self_supervised/moco/callbacks.py → .../models/self_supervised/moco/callbacks.py b/.../models/self_supervised/moco/callbacks.py → .../models/self_supervised/moco/callbacks.py
diff --git a/...dels/self_supervised/moco/moco2_module.py → ...dels/self_supervised/moco/moco2_module.py b/...dels/self_supervised/moco/moco2_module.py → ...dels/self_supervised/moco/moco2_module.py
diff --git a/...models/self_supervised/moco/transforms.py → ...models/self_supervised/moco/transforms.py b/...models/self_supervised/moco/transforms.py → ...models/self_supervised/moco/transforms.py
diff --git a/pl_bolts/models/self_supervised/resnets.py → ...l_bolts/models/self_supervised/resnets.py b/pl_bolts/models/self_supervised/resnets.py → ...l_bolts/models/self_supervised/resnets.py
diff --git a/...models/self_supervised/simclr/__init__.py → ...models/self_supervised/simclr/__init__.py b/...models/self_supervised/simclr/__init__.py → ...models/self_supervised/simclr/__init__.py
diff --git a/...elf_supervised/simclr/simclr_finetuner.py → ...elf_supervised/simclr/simclr_finetuner.py b/...elf_supervised/simclr/simclr_finetuner.py → ...elf_supervised/simclr/simclr_finetuner.py
diff --git a/...s/self_supervised/simclr/simclr_module.py → ...s/self_supervised/simclr/simclr_module.py b/...s/self_supervised/simclr/simclr_module.py → ...s/self_supervised/simclr/simclr_module.py
diff --git a/...dels/self_supervised/simclr/transforms.py → ...dels/self_supervised/simclr/transforms.py b/...dels/self_supervised/simclr/transforms.py → ...dels/self_supervised/simclr/transforms.py
diff --git a/...odels/self_supervised/simsiam/__init__.py → ...odels/self_supervised/simsiam/__init__.py b/...odels/self_supervised/simsiam/__init__.py → ...odels/self_supervised/simsiam/__init__.py
diff --git a/...self_supervised/simsiam/simsiam_module.py → ...self_supervised/simsiam/simsiam_module.py b/...self_supervised/simsiam/simsiam_module.py → ...self_supervised/simsiam/simsiam_module.py
diff --git a/...s/models/self_supervised/ssl_finetuner.py → ...s/models/self_supervised/ssl_finetuner.py b/...s/models/self_supervised/ssl_finetuner.py → ...s/models/self_supervised/ssl_finetuner.py
diff --git a/...s/models/self_supervised/swav/__init__.py → ...s/models/self_supervised/swav/__init__.py b/...s/models/self_supervised/swav/__init__.py → ...s/models/self_supervised/swav/__init__.py
diff --git a/pl_bolts/models/self_supervised/swav/loss.py → ...bolts/models/self_supervised/swav/loss.py b/pl_bolts/models/self_supervised/swav/loss.py → ...bolts/models/self_supervised/swav/loss.py
diff --git a/...ls/self_supervised/swav/swav_finetuner.py → ...ls/self_supervised/swav/swav_finetuner.py b/...ls/self_supervised/swav/swav_finetuner.py → ...ls/self_supervised/swav/swav_finetuner.py
diff --git a/...odels/self_supervised/swav/swav_module.py → ...odels/self_supervised/swav/swav_module.py b/...odels/self_supervised/swav/swav_module.py → ...odels/self_supervised/swav/swav_module.py
diff --git a/...odels/self_supervised/swav/swav_resnet.py → ...odels/self_supervised/swav/swav_resnet.py b/...odels/self_supervised/swav/swav_resnet.py → ...odels/self_supervised/swav/swav_resnet.py
diff --git a/...models/self_supervised/swav/transforms.py → ...models/self_supervised/swav/transforms.py b/...models/self_supervised/swav/transforms.py → ...models/self_supervised/swav/transforms.py
diff --git a/pl_bolts/models/vision/__init__.py → src/pl_bolts/models/vision/__init__.py b/pl_bolts/models/vision/__init__.py → src/pl_bolts/models/vision/__init__.py
diff --git a/pl_bolts/models/vision/image_gpt/__init__.py → ...bolts/models/vision/image_gpt/__init__.py b/pl_bolts/models/vision/image_gpt/__init__.py → ...bolts/models/vision/image_gpt/__init__.py
diff --git a/pl_bolts/models/vision/image_gpt/gpt2.py → src/pl_bolts/models/vision/image_gpt/gpt2.py b/pl_bolts/models/vision/image_gpt/gpt2.py → src/pl_bolts/models/vision/image_gpt/gpt2.py
diff --git a/...ts/models/vision/image_gpt/igpt_module.py → ...ts/models/vision/image_gpt/igpt_module.py b/...ts/models/vision/image_gpt/igpt_module.py → ...ts/models/vision/image_gpt/igpt_module.py
diff --git a/pl_bolts/models/vision/pixel_cnn.py → src/pl_bolts/models/vision/pixel_cnn.py b/pl_bolts/models/vision/pixel_cnn.py → src/pl_bolts/models/vision/pixel_cnn.py
diff --git a/pl_bolts/models/vision/segmentation.py → src/pl_bolts/models/vision/segmentation.py b/pl_bolts/models/vision/segmentation.py → src/pl_bolts/models/vision/segmentation.py
diff --git a/pl_bolts/models/vision/unet.py → src/pl_bolts/models/vision/unet.py b/pl_bolts/models/vision/unet.py → src/pl_bolts/models/vision/unet.py
diff --git a/pl_bolts/optimizers/__init__.py → src/pl_bolts/optimizers/__init__.py b/pl_bolts/optimizers/__init__.py → src/pl_bolts/optimizers/__init__.py
diff --git a/pl_bolts/optimizers/lars.py → src/pl_bolts/optimizers/lars.py b/pl_bolts/optimizers/lars.py → src/pl_bolts/optimizers/lars.py
diff --git a/pl_bolts/optimizers/lr_scheduler.py → src/pl_bolts/optimizers/lr_scheduler.py b/pl_bolts/optimizers/lr_scheduler.py → src/pl_bolts/optimizers/lr_scheduler.py
diff --git a/pl_bolts/transforms/__init__.py → src/pl_bolts/transforms/__init__.py b/pl_bolts/transforms/__init__.py → src/pl_bolts/transforms/__init__.py
diff --git a/...olts/transforms/dataset_normalizations.py → ...olts/transforms/dataset_normalizations.py b/...olts/transforms/dataset_normalizations.py → ...olts/transforms/dataset_normalizations.py
diff --git a/...ts/transforms/self_supervised/__init__.py → ...ts/transforms/self_supervised/__init__.py b/...ts/transforms/self_supervised/__init__.py → ...ts/transforms/self_supervised/__init__.py
diff --git a/...nsforms/self_supervised/ssl_transforms.py → ...nsforms/self_supervised/ssl_transforms.py b/...nsforms/self_supervised/ssl_transforms.py → ...nsforms/self_supervised/ssl_transforms.py
diff --git a/pl_bolts/utils/__init__.py → src/pl_bolts/utils/__init__.py b/pl_bolts/utils/__init__.py → src/pl_bolts/utils/__init__.py
diff --git a/pl_bolts/utils/arguments.py → src/pl_bolts/utils/arguments.py b/pl_bolts/utils/arguments.py → src/pl_bolts/utils/arguments.py
diff --git a/pl_bolts/utils/pretrained_weights.py → src/pl_bolts/utils/pretrained_weights.py b/pl_bolts/utils/pretrained_weights.py → src/pl_bolts/utils/pretrained_weights.py
diff --git a/pl_bolts/utils/self_supervised.py → src/pl_bolts/utils/self_supervised.py b/pl_bolts/utils/self_supervised.py → src/pl_bolts/utils/self_supervised.py
diff --git a/pl_bolts/utils/semi_supervised.py → src/pl_bolts/utils/semi_supervised.py b/pl_bolts/utils/semi_supervised.py → src/pl_bolts/utils/semi_supervised.py
diff --git a/pl_bolts/utils/shaping.py → src/pl_bolts/utils/shaping.py b/pl_bolts/utils/shaping.py → src/pl_bolts/utils/shaping.py
diff --git a/pl_bolts/utils/stability.py → src/pl_bolts/utils/stability.py b/pl_bolts/utils/stability.py → src/pl_bolts/utils/stability.py
diff --git a/pl_bolts/utils/types.py → src/pl_bolts/utils/types.py b/pl_bolts/utils/types.py → src/pl_bolts/utils/types.py
diff --git a/pl_bolts/utils/warnings.py → src/pl_bolts/utils/warnings.py b/pl_bolts/utils/warnings.py → src/pl_bolts/utils/warnings.py