├── .gitignore ├── .pre-commit-config.yaml ├── README.md ├── configs ├── callbacks │ ├── default.yaml │ └── none.yaml ├── datamodule │ ├── mnist.yaml │ └── prototype.yaml ├── debug │ ├── default.yaml │ ├── limit_batches.yaml │ ├── overfit.yaml │ ├── profiler.yaml │ ├── step.yaml │ └── test_only.yaml ├── hparams_search │ └── mnist_optuna.yaml ├── local │ └── .gitkeep ├── log_dir │ ├── debug.yaml │ ├── default.yaml │ └── evaluation.yaml ├── logger │ ├── tensorboard.yaml │ ├── wandb.yaml │ ├── wandb_debug.yaml │ └── wandb_eval.yaml ├── model │ ├── byol.yaml │ ├── mnist.yaml │ └── protonet.yaml ├── test.yaml ├── train.yaml └── trainer │ ├── ddp.yaml │ ├── default.yaml │ ├── train.yaml │ ├── with_eval.yaml │ └── without_eval.yaml ├── data_preparation.py ├── environment.yml ├── log.txt ├── pics ├── 12821655404104_.pic.jpg ├── 13071656977425_.pic.jpg └── VM.png ├── run.sh ├── search.py ├── setup.cfg ├── src ├── __init__.py ├── datamodules │ ├── __init__.py │ ├── components │ │ ├── Datagenerator.py │ │ ├── __init__.py │ │ ├── batch_sampler.py │ │ ├── dynamic_dataset.py │ │ ├── dynamic_pcen_dataset.py │ │ ├── dynamic_pcen_dataset_first_5.py │ │ ├── dynamic_pcen_dataset_val.py │ │ ├── identity_sampler.py │ │ ├── pcen.py │ │ ├── sisnr.py │ │ ├── test_loader.py │ │ ├── test_loader_ada_seglen.py │ │ └── test_loader_ada_seglen_better_neg_v2.py │ └── prototype_datamodule.py ├── models │ ├── __init__.py │ ├── components │ │ ├── __init__.py │ │ ├── byol_a.py │ │ ├── protonet.py │ │ ├── protonet_old.py │ │ └── simple_dense_net.py │ ├── eval_meta │ │ ├── subset_gt.csv │ │ └── subset_meta.csv │ └── prototype_module.py ├── testing_pipeline.py ├── training_pipeline.py ├── utils │ ├── __init__.py │ ├── evaluation.py │ ├── loss.py │ ├── metrics.py │ ├── post_proc.py │ ├── post_proc_new.py │ ├── pred.csv │ ├── ref.csv │ └── supconloss.py └── vendor │ └── __init__.py ├── sweep.yaml ├── test.py ├── tests ├── __init__.py ├── helpers │ ├── __init__.py │ ├── module_available.py │ ├── run_command.py │ └── runif.py ├── shell │ ├── __init__.py │ ├── test_basic_commands.py │ ├── test_debug_configs.py │ └── test_sweeps.py └── unit │ ├── __init__.py │ └── test_mnist_datamodule.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/README.md -------------------------------------------------------------------------------- /configs/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/callbacks/default.yaml -------------------------------------------------------------------------------- /configs/callbacks/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/datamodule/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/datamodule/mnist.yaml -------------------------------------------------------------------------------- /configs/datamodule/prototype.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/datamodule/prototype.yaml -------------------------------------------------------------------------------- /configs/debug/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/debug/default.yaml -------------------------------------------------------------------------------- /configs/debug/limit_batches.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/debug/limit_batches.yaml -------------------------------------------------------------------------------- /configs/debug/overfit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/debug/overfit.yaml -------------------------------------------------------------------------------- /configs/debug/profiler.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/debug/profiler.yaml -------------------------------------------------------------------------------- /configs/debug/step.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/debug/step.yaml -------------------------------------------------------------------------------- /configs/debug/test_only.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/debug/test_only.yaml -------------------------------------------------------------------------------- /configs/hparams_search/mnist_optuna.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/hparams_search/mnist_optuna.yaml -------------------------------------------------------------------------------- /configs/local/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/log_dir/debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/log_dir/debug.yaml -------------------------------------------------------------------------------- /configs/log_dir/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/log_dir/default.yaml -------------------------------------------------------------------------------- /configs/log_dir/evaluation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/log_dir/evaluation.yaml -------------------------------------------------------------------------------- /configs/logger/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/logger/tensorboard.yaml -------------------------------------------------------------------------------- /configs/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/logger/wandb.yaml -------------------------------------------------------------------------------- /configs/logger/wandb_debug.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/logger/wandb_debug.yaml -------------------------------------------------------------------------------- /configs/logger/wandb_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/logger/wandb_eval.yaml -------------------------------------------------------------------------------- /configs/model/byol.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/model/byol.yaml -------------------------------------------------------------------------------- /configs/model/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/model/mnist.yaml -------------------------------------------------------------------------------- /configs/model/protonet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/model/protonet.yaml -------------------------------------------------------------------------------- /configs/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/test.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/trainer/ddp.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /configs/trainer/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/trainer/train.yaml -------------------------------------------------------------------------------- /configs/trainer/with_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/trainer/with_eval.yaml -------------------------------------------------------------------------------- /configs/trainer/without_eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/configs/trainer/without_eval.yaml -------------------------------------------------------------------------------- /data_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/data_preparation.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/environment.yml -------------------------------------------------------------------------------- /log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/log.txt -------------------------------------------------------------------------------- /pics/12821655404104_.pic.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/pics/12821655404104_.pic.jpg -------------------------------------------------------------------------------- /pics/13071656977425_.pic.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/pics/13071656977425_.pic.jpg -------------------------------------------------------------------------------- /pics/VM.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/pics/VM.png -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/run.sh -------------------------------------------------------------------------------- /search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/search.py -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/setup.cfg -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datamodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datamodules/components/Datagenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/Datagenerator.py -------------------------------------------------------------------------------- /src/datamodules/components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/__init__.py -------------------------------------------------------------------------------- /src/datamodules/components/batch_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/batch_sampler.py -------------------------------------------------------------------------------- /src/datamodules/components/dynamic_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/dynamic_dataset.py -------------------------------------------------------------------------------- /src/datamodules/components/dynamic_pcen_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/dynamic_pcen_dataset.py -------------------------------------------------------------------------------- /src/datamodules/components/dynamic_pcen_dataset_first_5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/dynamic_pcen_dataset_first_5.py -------------------------------------------------------------------------------- /src/datamodules/components/dynamic_pcen_dataset_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/dynamic_pcen_dataset_val.py -------------------------------------------------------------------------------- /src/datamodules/components/identity_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/identity_sampler.py -------------------------------------------------------------------------------- /src/datamodules/components/pcen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/pcen.py -------------------------------------------------------------------------------- /src/datamodules/components/sisnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/sisnr.py -------------------------------------------------------------------------------- /src/datamodules/components/test_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/test_loader.py -------------------------------------------------------------------------------- /src/datamodules/components/test_loader_ada_seglen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/test_loader_ada_seglen.py -------------------------------------------------------------------------------- /src/datamodules/components/test_loader_ada_seglen_better_neg_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/components/test_loader_ada_seglen_better_neg_v2.py -------------------------------------------------------------------------------- /src/datamodules/prototype_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/datamodules/prototype_datamodule.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/byol_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/models/components/byol_a.py -------------------------------------------------------------------------------- /src/models/components/protonet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/models/components/protonet.py -------------------------------------------------------------------------------- /src/models/components/protonet_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/models/components/protonet_old.py -------------------------------------------------------------------------------- /src/models/components/simple_dense_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/models/components/simple_dense_net.py -------------------------------------------------------------------------------- /src/models/eval_meta/subset_gt.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/models/eval_meta/subset_gt.csv -------------------------------------------------------------------------------- /src/models/eval_meta/subset_meta.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/models/eval_meta/subset_meta.csv -------------------------------------------------------------------------------- /src/models/prototype_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/models/prototype_module.py -------------------------------------------------------------------------------- /src/testing_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/testing_pipeline.py -------------------------------------------------------------------------------- /src/training_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/training_pipeline.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/utils/evaluation.py -------------------------------------------------------------------------------- /src/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/utils/loss.py -------------------------------------------------------------------------------- /src/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/utils/metrics.py -------------------------------------------------------------------------------- /src/utils/post_proc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/utils/post_proc.py -------------------------------------------------------------------------------- /src/utils/post_proc_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/utils/post_proc_new.py -------------------------------------------------------------------------------- /src/utils/pred.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/utils/pred.csv -------------------------------------------------------------------------------- /src/utils/ref.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/utils/ref.csv -------------------------------------------------------------------------------- /src/utils/supconloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/utils/supconloss.py -------------------------------------------------------------------------------- /src/vendor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/src/vendor/__init__.py -------------------------------------------------------------------------------- /sweep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/sweep.yaml -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/test.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/helpers/module_available.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/tests/helpers/module_available.py -------------------------------------------------------------------------------- /tests/helpers/run_command.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/tests/helpers/run_command.py -------------------------------------------------------------------------------- /tests/helpers/runif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/tests/helpers/runif.py -------------------------------------------------------------------------------- /tests/shell/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/shell/test_basic_commands.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/tests/shell/test_basic_commands.py -------------------------------------------------------------------------------- /tests/shell/test_debug_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/tests/shell/test_debug_configs.py -------------------------------------------------------------------------------- /tests/shell/test_sweeps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/tests/shell/test_sweeps.py -------------------------------------------------------------------------------- /tests/unit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unit/test_mnist_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/tests/unit/test_mnist_datamodule.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoheliu/DCASE_2022_Task_5/HEAD/train.py --------------------------------------------------------------------------------