├── .gitignore ├── .pylintrc ├── README.md ├── core ├── __init__.py ├── datasets │ ├── __init__.py │ ├── cifar_lt_dataset.py │ ├── gtsrb_dataset.py │ └── kdatasets.py ├── ldam_drw │ ├── __init__.py │ ├── configs │ │ ├── cifar100_0.01.yaml │ │ ├── cifar10_0.01.yaml │ │ ├── k49bglt-2.0-munit.yaml │ │ ├── k49bglt-2.0.yaml │ │ ├── k49dillt-2.0-munit.yaml │ │ ├── k49dillt-2.0.yaml │ │ └── k49rotlt-2.0.yaml │ ├── losses.py │ ├── parse_arguments.py │ ├── source.md │ └── utils.py ├── make_kmnist_lt.py ├── models │ ├── __init__.py │ └── cifar_resnet.py ├── munit │ ├── __init__.py │ ├── configs │ │ ├── cifar100_0.01.yaml │ │ ├── cifar10_0.01.yaml │ │ ├── k49bglt-2.0.yaml │ │ └── k49dillt-2.0.yaml │ ├── data.py │ ├── munit_loading_utils.py │ ├── networks.py │ ├── trainer.py │ ├── utils.py │ └── viz.py ├── train_baselines.py ├── train_munit.py └── utils │ ├── __init__.py │ ├── common.py │ └── dataset_utils.py ├── environment.yml ├── pyproject.toml └── scripts ├── train_k49.sh └── train_k49_munits.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/.pylintrc -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/README.md -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/datasets/cifar_lt_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/datasets/cifar_lt_dataset.py -------------------------------------------------------------------------------- /core/datasets/gtsrb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/datasets/gtsrb_dataset.py -------------------------------------------------------------------------------- /core/datasets/kdatasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/datasets/kdatasets.py -------------------------------------------------------------------------------- /core/ldam_drw/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/ldam_drw/configs/cifar100_0.01.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/configs/cifar100_0.01.yaml -------------------------------------------------------------------------------- /core/ldam_drw/configs/cifar10_0.01.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/configs/cifar10_0.01.yaml -------------------------------------------------------------------------------- /core/ldam_drw/configs/k49bglt-2.0-munit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/configs/k49bglt-2.0-munit.yaml -------------------------------------------------------------------------------- /core/ldam_drw/configs/k49bglt-2.0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/configs/k49bglt-2.0.yaml -------------------------------------------------------------------------------- /core/ldam_drw/configs/k49dillt-2.0-munit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/configs/k49dillt-2.0-munit.yaml -------------------------------------------------------------------------------- /core/ldam_drw/configs/k49dillt-2.0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/configs/k49dillt-2.0.yaml -------------------------------------------------------------------------------- /core/ldam_drw/configs/k49rotlt-2.0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/configs/k49rotlt-2.0.yaml -------------------------------------------------------------------------------- /core/ldam_drw/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/losses.py -------------------------------------------------------------------------------- /core/ldam_drw/parse_arguments.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/parse_arguments.py -------------------------------------------------------------------------------- /core/ldam_drw/source.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/source.md -------------------------------------------------------------------------------- /core/ldam_drw/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/ldam_drw/utils.py -------------------------------------------------------------------------------- /core/make_kmnist_lt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/make_kmnist_lt.py -------------------------------------------------------------------------------- /core/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/models/cifar_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/models/cifar_resnet.py -------------------------------------------------------------------------------- /core/munit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/munit/configs/cifar100_0.01.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/configs/cifar100_0.01.yaml -------------------------------------------------------------------------------- /core/munit/configs/cifar10_0.01.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/configs/cifar10_0.01.yaml -------------------------------------------------------------------------------- /core/munit/configs/k49bglt-2.0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/configs/k49bglt-2.0.yaml -------------------------------------------------------------------------------- /core/munit/configs/k49dillt-2.0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/configs/k49dillt-2.0.yaml -------------------------------------------------------------------------------- /core/munit/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/data.py -------------------------------------------------------------------------------- /core/munit/munit_loading_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/munit_loading_utils.py -------------------------------------------------------------------------------- /core/munit/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/networks.py -------------------------------------------------------------------------------- /core/munit/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/trainer.py -------------------------------------------------------------------------------- /core/munit/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/utils.py -------------------------------------------------------------------------------- /core/munit/viz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/munit/viz.py -------------------------------------------------------------------------------- /core/train_baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/train_baselines.py -------------------------------------------------------------------------------- /core/train_munit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/train_munit.py -------------------------------------------------------------------------------- /core/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/utils/common.py -------------------------------------------------------------------------------- /core/utils/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/core/utils/dataset_utils.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/environment.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- 1 | [tool.black] 2 | line-length = 95 3 | -------------------------------------------------------------------------------- /scripts/train_k49.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/scripts/train_k49.sh -------------------------------------------------------------------------------- /scripts/train_k49_munits.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AllanYangZhou/generative-invariance-transfer/HEAD/scripts/train_k49_munits.sh --------------------------------------------------------------------------------