├── LICENSE.txt ├── README.md ├── assets └── icon.png ├── code ├── LICENSE ├── __init__.py ├── application_sample.py ├── classifier_rejection.py ├── cm │ ├── __init__.py │ ├── classifier.py │ ├── dist_util.py │ ├── enc_dec_lib.py │ ├── evaluator.py │ ├── fp16_util.py │ ├── image_datasets.py │ ├── karras_diffusion.py │ ├── logger.py │ ├── losses.py │ ├── networks.py │ ├── nn.py │ ├── random_util.py │ ├── resample.py │ ├── sample_util.py │ ├── script_util.py │ ├── train_util.py │ └── unet.py ├── cm_train.py ├── cmd │ └── ImageNet64 │ │ ├── FID │ │ ├── get_image.sh │ │ ├── sampling-AWS.sh │ │ ├── sampling.sh │ │ ├── sampling_.sh │ │ └── sampling_bash.sh │ │ ├── cmd-ABCI │ │ ├── GAN_default.sh │ │ ├── fine_tune_ver2 │ │ │ ├── M20_w1.sh │ │ │ ├── M30_w1.5.sh │ │ │ ├── M30_w1.sh │ │ │ ├── M39_w1.5.sh │ │ │ └── M39_w1.sh │ │ └── fine_tune_ver3 │ │ │ ├── M20_w0.75.sh │ │ │ ├── M20_w1.25.sh │ │ │ ├── M20_w1.5.sh │ │ │ ├── M20_w1.75.sh │ │ │ ├── M20_w1.sh │ │ │ ├── M30_w0.8.sh │ │ │ ├── M30_w1.2.sh │ │ │ ├── M30_w1.sh │ │ │ ├── M39_w0.8.sh │ │ │ ├── M39_w1.2.sh │ │ │ └── M39_w1.sh │ │ ├── cmd-AWS │ │ ├── GAN.sh │ │ ├── M20_w1.sh │ │ └── test.sh │ │ └── cmd-GAIA │ │ ├── test-ver2.sh │ │ └── test.sh ├── datasets │ ├── README.md │ └── lsun_bedroom.py ├── dnnlib │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-311.pyc │ │ └── util.cpython-311.pyc │ └── util.py ├── eval_bpd.py ├── evaluations │ ├── __init__.py │ ├── evaluator.py │ ├── evaluator_.py │ ├── from_npz_to_npz.py │ ├── inception_v3.py │ ├── requirements.txt │ └── th_evaluator.py ├── feature_networks │ ├── clip │ │ ├── __init__.py │ │ ├── bpe_simple_vocab_16e6.txt.gz │ │ ├── clip.py │ │ ├── model.py │ │ └── simple_tokenizer.py │ ├── constants.py │ ├── pretrained_builder.py │ └── vit.py ├── fid_npzs.py ├── image_sample.py ├── model-card.md ├── pg_modules │ ├── blocks.py │ ├── diffaug.py │ ├── discriminator.py │ ├── in_embeddings │ │ └── tf_efficientnet_lite0.pkl │ ├── networks_stylegan2.py │ └── projector.py ├── scripts │ ├── cm_train.py │ ├── edm_train.py │ ├── image_sample.py │ ├── launch.sh │ └── ternary_search.py ├── setup.py ├── torch_utils │ ├── __init__.py │ ├── custom_ops.py │ ├── misc.py │ ├── ops │ │ ├── __init__.py │ │ ├── bias_act.cpp │ │ ├── bias_act.cu │ │ ├── bias_act.h │ │ ├── bias_act.py │ │ ├── conv2d_gradfix.py │ │ ├── conv2d_resample.py │ │ ├── fma.py │ │ ├── grid_sample_gradfix.py │ │ ├── upfirdn2d.cpp │ │ ├── upfirdn2d.cu │ │ ├── upfirdn2d.h │ │ └── upfirdn2d.py │ ├── persistence.py │ └── training_stats.py └── torch_utils_cm │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-311.pyc │ ├── misc.cpython-311.pyc │ └── persistence.cpython-311.pyc │ ├── distributed.py │ ├── misc.py │ ├── persistence.py │ └── training_stats.py └── commands ├── CTM+DSM+GAN_command.sh ├── CTM+DSM_command.sh └── sampling_command.sh /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/README.md -------------------------------------------------------------------------------- /assets/icon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/assets/icon.png -------------------------------------------------------------------------------- /code/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/LICENSE -------------------------------------------------------------------------------- /code/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/application_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/application_sample.py -------------------------------------------------------------------------------- /code/classifier_rejection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/classifier_rejection.py -------------------------------------------------------------------------------- /code/cm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/__init__.py -------------------------------------------------------------------------------- /code/cm/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/classifier.py -------------------------------------------------------------------------------- /code/cm/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/dist_util.py -------------------------------------------------------------------------------- /code/cm/enc_dec_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/enc_dec_lib.py -------------------------------------------------------------------------------- /code/cm/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/evaluator.py -------------------------------------------------------------------------------- /code/cm/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/fp16_util.py -------------------------------------------------------------------------------- /code/cm/image_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/image_datasets.py -------------------------------------------------------------------------------- /code/cm/karras_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/karras_diffusion.py -------------------------------------------------------------------------------- /code/cm/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/logger.py -------------------------------------------------------------------------------- /code/cm/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/losses.py -------------------------------------------------------------------------------- /code/cm/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/networks.py -------------------------------------------------------------------------------- /code/cm/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/nn.py -------------------------------------------------------------------------------- /code/cm/random_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/random_util.py -------------------------------------------------------------------------------- /code/cm/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/resample.py -------------------------------------------------------------------------------- /code/cm/sample_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/sample_util.py -------------------------------------------------------------------------------- /code/cm/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/script_util.py -------------------------------------------------------------------------------- /code/cm/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/train_util.py -------------------------------------------------------------------------------- /code/cm/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm/unet.py -------------------------------------------------------------------------------- /code/cm_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cm_train.py -------------------------------------------------------------------------------- /code/cmd/ImageNet64/FID/get_image.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/FID/get_image.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/FID/sampling-AWS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/FID/sampling-AWS.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/FID/sampling.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/FID/sampling.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/FID/sampling_.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/FID/sampling_.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/FID/sampling_bash.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/FID/sampling_bash.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/GAN_default.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/GAN_default.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M20_w1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M20_w1.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M30_w1.5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M30_w1.5.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M30_w1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M30_w1.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M39_w1.5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M39_w1.5.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M39_w1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver2/M39_w1.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w0.75.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w0.75.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w1.25.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w1.25.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w1.5.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w1.5.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w1.75.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w1.75.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M20_w1.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M30_w0.8.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M30_w0.8.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M30_w1.2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M30_w1.2.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M30_w1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M30_w1.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M39_w0.8.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M39_w0.8.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M39_w1.2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M39_w1.2.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M39_w1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-ABCI/fine_tune_ver3/M39_w1.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-AWS/GAN.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-AWS/GAN.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-AWS/M20_w1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-AWS/M20_w1.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-AWS/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-AWS/test.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-GAIA/test-ver2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-GAIA/test-ver2.sh -------------------------------------------------------------------------------- /code/cmd/ImageNet64/cmd-GAIA/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/cmd/ImageNet64/cmd-GAIA/test.sh -------------------------------------------------------------------------------- /code/datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/datasets/README.md -------------------------------------------------------------------------------- /code/datasets/lsun_bedroom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/datasets/lsun_bedroom.py -------------------------------------------------------------------------------- /code/dnnlib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/dnnlib/__init__.py -------------------------------------------------------------------------------- /code/dnnlib/__pycache__/__init__.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/dnnlib/__pycache__/__init__.cpython-311.pyc -------------------------------------------------------------------------------- /code/dnnlib/__pycache__/util.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/dnnlib/__pycache__/util.cpython-311.pyc -------------------------------------------------------------------------------- /code/dnnlib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/dnnlib/util.py -------------------------------------------------------------------------------- /code/eval_bpd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/eval_bpd.py -------------------------------------------------------------------------------- /code/evaluations/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/evaluations/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/evaluations/evaluator.py -------------------------------------------------------------------------------- /code/evaluations/evaluator_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/evaluations/evaluator_.py -------------------------------------------------------------------------------- /code/evaluations/from_npz_to_npz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/evaluations/from_npz_to_npz.py -------------------------------------------------------------------------------- /code/evaluations/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/evaluations/inception_v3.py -------------------------------------------------------------------------------- /code/evaluations/requirements.txt: -------------------------------------------------------------------------------- 1 | tensorflow-gpu>=2.0 2 | scipy 3 | requests 4 | tqdm -------------------------------------------------------------------------------- /code/evaluations/th_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/evaluations/th_evaluator.py -------------------------------------------------------------------------------- /code/feature_networks/clip/__init__.py: -------------------------------------------------------------------------------- 1 | from .clip import * 2 | -------------------------------------------------------------------------------- /code/feature_networks/clip/bpe_simple_vocab_16e6.txt.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/feature_networks/clip/bpe_simple_vocab_16e6.txt.gz -------------------------------------------------------------------------------- /code/feature_networks/clip/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/feature_networks/clip/clip.py -------------------------------------------------------------------------------- /code/feature_networks/clip/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/feature_networks/clip/model.py -------------------------------------------------------------------------------- /code/feature_networks/clip/simple_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/feature_networks/clip/simple_tokenizer.py -------------------------------------------------------------------------------- /code/feature_networks/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/feature_networks/constants.py -------------------------------------------------------------------------------- /code/feature_networks/pretrained_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/feature_networks/pretrained_builder.py -------------------------------------------------------------------------------- /code/feature_networks/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/feature_networks/vit.py -------------------------------------------------------------------------------- /code/fid_npzs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/fid_npzs.py -------------------------------------------------------------------------------- /code/image_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/image_sample.py -------------------------------------------------------------------------------- /code/model-card.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/model-card.md -------------------------------------------------------------------------------- /code/pg_modules/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/pg_modules/blocks.py -------------------------------------------------------------------------------- /code/pg_modules/diffaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/pg_modules/diffaug.py -------------------------------------------------------------------------------- /code/pg_modules/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/pg_modules/discriminator.py -------------------------------------------------------------------------------- /code/pg_modules/in_embeddings/tf_efficientnet_lite0.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/pg_modules/in_embeddings/tf_efficientnet_lite0.pkl -------------------------------------------------------------------------------- /code/pg_modules/networks_stylegan2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/pg_modules/networks_stylegan2.py -------------------------------------------------------------------------------- /code/pg_modules/projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/pg_modules/projector.py -------------------------------------------------------------------------------- /code/scripts/cm_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/scripts/cm_train.py -------------------------------------------------------------------------------- /code/scripts/edm_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/scripts/edm_train.py -------------------------------------------------------------------------------- /code/scripts/image_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/scripts/image_sample.py -------------------------------------------------------------------------------- /code/scripts/launch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/scripts/launch.sh -------------------------------------------------------------------------------- /code/scripts/ternary_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/scripts/ternary_search.py -------------------------------------------------------------------------------- /code/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/setup.py -------------------------------------------------------------------------------- /code/torch_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/__init__.py -------------------------------------------------------------------------------- /code/torch_utils/custom_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/custom_ops.py -------------------------------------------------------------------------------- /code/torch_utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/misc.py -------------------------------------------------------------------------------- /code/torch_utils/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/__init__.py -------------------------------------------------------------------------------- /code/torch_utils/ops/bias_act.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/bias_act.cpp -------------------------------------------------------------------------------- /code/torch_utils/ops/bias_act.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/bias_act.cu -------------------------------------------------------------------------------- /code/torch_utils/ops/bias_act.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/bias_act.h -------------------------------------------------------------------------------- /code/torch_utils/ops/bias_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/bias_act.py -------------------------------------------------------------------------------- /code/torch_utils/ops/conv2d_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/conv2d_gradfix.py -------------------------------------------------------------------------------- /code/torch_utils/ops/conv2d_resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/conv2d_resample.py -------------------------------------------------------------------------------- /code/torch_utils/ops/fma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/fma.py -------------------------------------------------------------------------------- /code/torch_utils/ops/grid_sample_gradfix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/grid_sample_gradfix.py -------------------------------------------------------------------------------- /code/torch_utils/ops/upfirdn2d.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/upfirdn2d.cpp -------------------------------------------------------------------------------- /code/torch_utils/ops/upfirdn2d.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/upfirdn2d.cu -------------------------------------------------------------------------------- /code/torch_utils/ops/upfirdn2d.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/upfirdn2d.h -------------------------------------------------------------------------------- /code/torch_utils/ops/upfirdn2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/ops/upfirdn2d.py -------------------------------------------------------------------------------- /code/torch_utils/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/persistence.py -------------------------------------------------------------------------------- /code/torch_utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils/training_stats.py -------------------------------------------------------------------------------- /code/torch_utils_cm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils_cm/__init__.py -------------------------------------------------------------------------------- /code/torch_utils_cm/__pycache__/__init__.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils_cm/__pycache__/__init__.cpython-311.pyc -------------------------------------------------------------------------------- /code/torch_utils_cm/__pycache__/misc.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils_cm/__pycache__/misc.cpython-311.pyc -------------------------------------------------------------------------------- /code/torch_utils_cm/__pycache__/persistence.cpython-311.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils_cm/__pycache__/persistence.cpython-311.pyc -------------------------------------------------------------------------------- /code/torch_utils_cm/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils_cm/distributed.py -------------------------------------------------------------------------------- /code/torch_utils_cm/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils_cm/misc.py -------------------------------------------------------------------------------- /code/torch_utils_cm/persistence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils_cm/persistence.py -------------------------------------------------------------------------------- /code/torch_utils_cm/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/code/torch_utils_cm/training_stats.py -------------------------------------------------------------------------------- /commands/CTM+DSM+GAN_command.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/commands/CTM+DSM+GAN_command.sh -------------------------------------------------------------------------------- /commands/CTM+DSM_command.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/commands/CTM+DSM_command.sh -------------------------------------------------------------------------------- /commands/sampling_command.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sony/ctm/HEAD/commands/sampling_command.sh --------------------------------------------------------------------------------