├── .env.example ├── .gitignore ├── .pre-commit-config.yaml ├── .project-root ├── Imaging.ipynb ├── ImagingModel.csv ├── LICENSE ├── Makefile ├── README.md ├── configs ├── __init__.py ├── callbacks │ ├── default.yaml │ ├── early_stopping.yaml │ ├── model_checkpoint.yaml │ ├── model_summary.yaml │ ├── none.yaml │ └── rich_progress_bar.yaml ├── data │ └── mnist.yaml ├── debug │ ├── default.yaml │ ├── fdr.yaml │ ├── limit.yaml │ ├── overfit.yaml │ └── profiler.yaml ├── eval.yaml ├── experiment │ └── example.yaml ├── extras │ └── default.yaml ├── hparams_search │ └── mnist_optuna.yaml ├── hydra │ └── default.yaml ├── local │ └── .gitkeep ├── logger │ ├── aim.yaml │ ├── comet.yaml │ ├── csv.yaml │ ├── many_loggers.yaml │ ├── mlflow.yaml │ ├── neptune.yaml │ ├── tensorboard.yaml │ └── wandb.yaml ├── model │ └── mnist.yaml ├── paths │ └── default.yaml ├── train.yaml └── trainer │ ├── cpu.yaml │ ├── ddp.yaml │ ├── ddp_sim.yaml │ ├── default.yaml │ ├── gpu.yaml │ └── mps.yaml ├── environment.yaml ├── mask.csv ├── misc ├── figs │ ├── torchlitho.pdf │ ├── torchlitho.png │ └── torchlitho │ │ ├── social_preview.png │ │ └── torchlitho.png └── slides │ └── torchlitho.pdf ├── notebooks └── .gitkeep ├── pyproject.toml ├── requirements.txt ├── scripts └── schedule.sh ├── setup.py ├── src ├── __init__.py ├── data │ ├── __init__.py │ ├── components │ │ └── __init__.py │ └── mnist_datamodule.py ├── eval.py ├── models │ ├── __init__.py │ ├── components │ │ ├── __init__.py │ │ └── simple_dense_net.py │ ├── gdsii │ │ ├── __init__.py │ │ ├── _records.py │ │ ├── elements.py │ │ ├── exceptions.py │ │ ├── library.py │ │ ├── record.py │ │ ├── structure.py │ │ ├── tags.py │ │ └── types.py │ ├── litho │ │ ├── FilmStack.py │ │ ├── ImageData.py │ │ ├── ImagingModel.py │ │ ├── ImagingView.py │ │ ├── Layer.py │ │ ├── Mask.py │ │ ├── Masks.py │ │ ├── Material.py │ │ ├── Numerics.py │ │ ├── ProjectionObjective.py │ │ ├── Receipe.py │ │ ├── Resist.py │ │ ├── Source.py │ │ ├── __init__.py │ │ └── functions │ │ │ ├── Calculate1DAerialImage.py │ │ │ ├── Calculate1DResistImage.py │ │ │ ├── Calculate2DAerialImage.py │ │ │ ├── Calculate2DResistImage.py │ │ │ ├── Calculate2DTCCMatrix.py │ │ │ ├── CalculateAerialImage_SOCS.py │ │ │ ├── CalculateCharacteristicMatrix.py │ │ │ ├── CalculateNormalImage.py │ │ │ ├── CalculateTransferMatrix.py │ │ │ ├── DecomposeTCC_SOCS.py │ │ │ └── __init__.py │ └── mnist_module.py ├── train.py └── utils │ ├── __init__.py │ ├── instantiators.py │ ├── logging_utils.py │ ├── pylogger.py │ ├── rich_utils.py │ └── utils.py └── tests ├── __init__.py ├── conftest.py ├── helpers ├── __init__.py ├── package_available.py ├── run_if.py └── run_sh_command.py ├── test_configs.py ├── test_datamodules.py ├── test_eval.py ├── test_sweeps.py └── test_train.py /.env.example: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/.env.example -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.project-root: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/.project-root -------------------------------------------------------------------------------- /Imaging.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/Imaging.ipynb -------------------------------------------------------------------------------- /ImagingModel.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/ImagingModel.csv -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/Makefile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/__init__.py -------------------------------------------------------------------------------- /configs/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/callbacks/default.yaml -------------------------------------------------------------------------------- /configs/callbacks/early_stopping.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/callbacks/early_stopping.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_checkpoint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/callbacks/model_checkpoint.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_summary.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/callbacks/model_summary.yaml -------------------------------------------------------------------------------- /configs/callbacks/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/callbacks/rich_progress_bar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/callbacks/rich_progress_bar.yaml -------------------------------------------------------------------------------- /configs/data/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/data/mnist.yaml -------------------------------------------------------------------------------- /configs/debug/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/debug/default.yaml -------------------------------------------------------------------------------- /configs/debug/fdr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/debug/fdr.yaml -------------------------------------------------------------------------------- /configs/debug/limit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/debug/limit.yaml -------------------------------------------------------------------------------- /configs/debug/overfit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/debug/overfit.yaml -------------------------------------------------------------------------------- /configs/debug/profiler.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/debug/profiler.yaml -------------------------------------------------------------------------------- /configs/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/eval.yaml -------------------------------------------------------------------------------- /configs/experiment/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/experiment/example.yaml -------------------------------------------------------------------------------- /configs/extras/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/extras/default.yaml -------------------------------------------------------------------------------- /configs/hparams_search/mnist_optuna.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/hparams_search/mnist_optuna.yaml -------------------------------------------------------------------------------- /configs/hydra/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/hydra/default.yaml -------------------------------------------------------------------------------- /configs/local/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/logger/aim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/logger/aim.yaml -------------------------------------------------------------------------------- /configs/logger/comet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/logger/comet.yaml -------------------------------------------------------------------------------- /configs/logger/csv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/logger/csv.yaml -------------------------------------------------------------------------------- /configs/logger/many_loggers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/logger/many_loggers.yaml -------------------------------------------------------------------------------- /configs/logger/mlflow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/logger/mlflow.yaml -------------------------------------------------------------------------------- /configs/logger/neptune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/logger/neptune.yaml -------------------------------------------------------------------------------- /configs/logger/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/logger/tensorboard.yaml -------------------------------------------------------------------------------- /configs/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/logger/wandb.yaml -------------------------------------------------------------------------------- /configs/model/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/model/mnist.yaml -------------------------------------------------------------------------------- /configs/paths/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/paths/default.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/trainer/cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/trainer/cpu.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/trainer/ddp.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp_sim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/trainer/ddp_sim.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /configs/trainer/gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/trainer/gpu.yaml -------------------------------------------------------------------------------- /configs/trainer/mps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/configs/trainer/mps.yaml -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/environment.yaml -------------------------------------------------------------------------------- /mask.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/mask.csv -------------------------------------------------------------------------------- /misc/figs/torchlitho.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/misc/figs/torchlitho.pdf -------------------------------------------------------------------------------- /misc/figs/torchlitho.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/misc/figs/torchlitho.png -------------------------------------------------------------------------------- /misc/figs/torchlitho/social_preview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/misc/figs/torchlitho/social_preview.png -------------------------------------------------------------------------------- /misc/figs/torchlitho/torchlitho.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/misc/figs/torchlitho/torchlitho.png -------------------------------------------------------------------------------- /misc/slides/torchlitho.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/misc/slides/torchlitho.pdf -------------------------------------------------------------------------------- /notebooks/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/schedule.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/scripts/schedule.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/setup.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/mnist_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/data/mnist_datamodule.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/simple_dense_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/components/simple_dense_net.py -------------------------------------------------------------------------------- /src/models/gdsii/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/gdsii/_records.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/gdsii/_records.py -------------------------------------------------------------------------------- /src/models/gdsii/elements.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/gdsii/elements.py -------------------------------------------------------------------------------- /src/models/gdsii/exceptions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/gdsii/exceptions.py -------------------------------------------------------------------------------- /src/models/gdsii/library.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/gdsii/library.py -------------------------------------------------------------------------------- /src/models/gdsii/record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/gdsii/record.py -------------------------------------------------------------------------------- /src/models/gdsii/structure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/gdsii/structure.py -------------------------------------------------------------------------------- /src/models/gdsii/tags.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/gdsii/tags.py -------------------------------------------------------------------------------- /src/models/gdsii/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/gdsii/types.py -------------------------------------------------------------------------------- /src/models/litho/FilmStack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/FilmStack.py -------------------------------------------------------------------------------- /src/models/litho/ImageData.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/ImageData.py -------------------------------------------------------------------------------- /src/models/litho/ImagingModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/ImagingModel.py -------------------------------------------------------------------------------- /src/models/litho/ImagingView.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/ImagingView.py -------------------------------------------------------------------------------- /src/models/litho/Layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/Layer.py -------------------------------------------------------------------------------- /src/models/litho/Mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/Mask.py -------------------------------------------------------------------------------- /src/models/litho/Masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/Masks.py -------------------------------------------------------------------------------- /src/models/litho/Material.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/Material.py -------------------------------------------------------------------------------- /src/models/litho/Numerics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/Numerics.py -------------------------------------------------------------------------------- /src/models/litho/ProjectionObjective.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/ProjectionObjective.py -------------------------------------------------------------------------------- /src/models/litho/Receipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/Receipe.py -------------------------------------------------------------------------------- /src/models/litho/Resist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/Resist.py -------------------------------------------------------------------------------- /src/models/litho/Source.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/Source.py -------------------------------------------------------------------------------- /src/models/litho/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/litho/functions/Calculate1DAerialImage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/Calculate1DAerialImage.py -------------------------------------------------------------------------------- /src/models/litho/functions/Calculate1DResistImage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/Calculate1DResistImage.py -------------------------------------------------------------------------------- /src/models/litho/functions/Calculate2DAerialImage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/Calculate2DAerialImage.py -------------------------------------------------------------------------------- /src/models/litho/functions/Calculate2DResistImage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/Calculate2DResistImage.py -------------------------------------------------------------------------------- /src/models/litho/functions/Calculate2DTCCMatrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/Calculate2DTCCMatrix.py -------------------------------------------------------------------------------- /src/models/litho/functions/CalculateAerialImage_SOCS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/CalculateAerialImage_SOCS.py -------------------------------------------------------------------------------- /src/models/litho/functions/CalculateCharacteristicMatrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/CalculateCharacteristicMatrix.py -------------------------------------------------------------------------------- /src/models/litho/functions/CalculateNormalImage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/CalculateNormalImage.py -------------------------------------------------------------------------------- /src/models/litho/functions/CalculateTransferMatrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/CalculateTransferMatrix.py -------------------------------------------------------------------------------- /src/models/litho/functions/DecomposeTCC_SOCS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/litho/functions/DecomposeTCC_SOCS.py -------------------------------------------------------------------------------- /src/models/litho/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/mnist_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/models/mnist_module.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/instantiators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/utils/instantiators.py -------------------------------------------------------------------------------- /src/utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/utils/logging_utils.py -------------------------------------------------------------------------------- /src/utils/pylogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/utils/pylogger.py -------------------------------------------------------------------------------- /src/utils/rich_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/utils/rich_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/src/utils/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/helpers/package_available.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/tests/helpers/package_available.py -------------------------------------------------------------------------------- /tests/helpers/run_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/tests/helpers/run_if.py -------------------------------------------------------------------------------- /tests/helpers/run_sh_command.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/tests/helpers/run_sh_command.py -------------------------------------------------------------------------------- /tests/test_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/tests/test_configs.py -------------------------------------------------------------------------------- /tests/test_datamodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/tests/test_datamodules.py -------------------------------------------------------------------------------- /tests/test_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/tests/test_eval.py -------------------------------------------------------------------------------- /tests/test_sweeps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/tests/test_sweeps.py -------------------------------------------------------------------------------- /tests/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TorchOPC/TorchLitho/HEAD/tests/test_train.py --------------------------------------------------------------------------------