├── .gitignore ├── LICENSE ├── README.md ├── Trainer.py ├── configs ├── data_loaders │ ├── sms_wsj.yaml │ ├── whamr.yaml │ └── wsj-2mix.yaml ├── lr_scheduler │ ├── CyclicLRExpRange.yaml │ ├── ExponentialLR.yaml │ ├── OneCycleLR.yaml │ ├── ReduceLROnPlateau.yaml │ └── WarmupReduceLROnPlateau.yaml └── models │ └── CrossNet.yaml ├── create_env.sh ├── data_loaders ├── __init__.py ├── sms_wsj.py ├── utils │ ├── array_geometry.py │ ├── augment.py │ ├── collate_func.py │ ├── diffuse_noise.py │ ├── mix.py │ ├── my_distributed_sampler.py │ ├── rand.py │ └── window.py ├── whamr.py └── wsj_2mix.py ├── models ├── __init__.py ├── arch │ ├── CrossNet.py │ ├── __init__.py │ └── base │ │ ├── linear_group.py │ │ ├── non_linear.py │ │ └── norm.py ├── io │ ├── __init__.py │ ├── loss.py │ ├── norm.py │ ├── pit.py │ └── stft.py ├── optim │ ├── __init__.py │ └── lr_scheduler.py └── utils │ ├── __init__.py │ ├── asr_metric.py │ ├── base_cli.py │ ├── ensemble.py │ ├── flops.py │ ├── general_steps.py │ ├── git_tools.py │ ├── metrics.py │ ├── my_json_encoder.py │ ├── my_logger.py │ ├── my_progress_bar.py │ ├── my_rich_progress_bar.py │ ├── my_save_config_callback.py │ ├── shared_cli.py │ └── snr_sch.py └── requirements.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/README.md -------------------------------------------------------------------------------- /Trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/Trainer.py -------------------------------------------------------------------------------- /configs/data_loaders/sms_wsj.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/configs/data_loaders/sms_wsj.yaml -------------------------------------------------------------------------------- /configs/data_loaders/whamr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/configs/data_loaders/whamr.yaml -------------------------------------------------------------------------------- /configs/data_loaders/wsj-2mix.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/configs/data_loaders/wsj-2mix.yaml -------------------------------------------------------------------------------- /configs/lr_scheduler/CyclicLRExpRange.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/configs/lr_scheduler/CyclicLRExpRange.yaml -------------------------------------------------------------------------------- /configs/lr_scheduler/ExponentialLR.yaml: -------------------------------------------------------------------------------- 1 | model: 2 | lr_scheduler: [ExponentialLR, { gamma: 0.99 }] 3 | -------------------------------------------------------------------------------- /configs/lr_scheduler/OneCycleLR.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/configs/lr_scheduler/OneCycleLR.yaml -------------------------------------------------------------------------------- /configs/lr_scheduler/ReduceLROnPlateau.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/configs/lr_scheduler/ReduceLROnPlateau.yaml -------------------------------------------------------------------------------- /configs/lr_scheduler/WarmupReduceLROnPlateau.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/configs/lr_scheduler/WarmupReduceLROnPlateau.yaml -------------------------------------------------------------------------------- /configs/models/CrossNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/configs/models/CrossNet.yaml -------------------------------------------------------------------------------- /create_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/create_env.sh -------------------------------------------------------------------------------- /data_loaders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/__init__.py -------------------------------------------------------------------------------- /data_loaders/sms_wsj.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/sms_wsj.py -------------------------------------------------------------------------------- /data_loaders/utils/array_geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/utils/array_geometry.py -------------------------------------------------------------------------------- /data_loaders/utils/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/utils/augment.py -------------------------------------------------------------------------------- /data_loaders/utils/collate_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/utils/collate_func.py -------------------------------------------------------------------------------- /data_loaders/utils/diffuse_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/utils/diffuse_noise.py -------------------------------------------------------------------------------- /data_loaders/utils/mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/utils/mix.py -------------------------------------------------------------------------------- /data_loaders/utils/my_distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/utils/my_distributed_sampler.py -------------------------------------------------------------------------------- /data_loaders/utils/rand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/utils/rand.py -------------------------------------------------------------------------------- /data_loaders/utils/window.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/utils/window.py -------------------------------------------------------------------------------- /data_loaders/whamr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/whamr.py -------------------------------------------------------------------------------- /data_loaders/wsj_2mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/data_loaders/wsj_2mix.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | # from SpatialNet repo -------------------------------------------------------------------------------- /models/arch/CrossNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/arch/CrossNet.py -------------------------------------------------------------------------------- /models/arch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/arch/base/linear_group.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/arch/base/linear_group.py -------------------------------------------------------------------------------- /models/arch/base/non_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/arch/base/non_linear.py -------------------------------------------------------------------------------- /models/arch/base/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/arch/base/norm.py -------------------------------------------------------------------------------- /models/io/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/io/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/io/loss.py -------------------------------------------------------------------------------- /models/io/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/io/norm.py -------------------------------------------------------------------------------- /models/io/pit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/io/pit.py -------------------------------------------------------------------------------- /models/io/stft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/io/stft.py -------------------------------------------------------------------------------- /models/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/optim/__init__.py -------------------------------------------------------------------------------- /models/optim/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/optim/lr_scheduler.py -------------------------------------------------------------------------------- /models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/__init__.py -------------------------------------------------------------------------------- /models/utils/asr_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/asr_metric.py -------------------------------------------------------------------------------- /models/utils/base_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/base_cli.py -------------------------------------------------------------------------------- /models/utils/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/ensemble.py -------------------------------------------------------------------------------- /models/utils/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/flops.py -------------------------------------------------------------------------------- /models/utils/general_steps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/general_steps.py -------------------------------------------------------------------------------- /models/utils/git_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/git_tools.py -------------------------------------------------------------------------------- /models/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/metrics.py -------------------------------------------------------------------------------- /models/utils/my_json_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/my_json_encoder.py -------------------------------------------------------------------------------- /models/utils/my_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/my_logger.py -------------------------------------------------------------------------------- /models/utils/my_progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/my_progress_bar.py -------------------------------------------------------------------------------- /models/utils/my_rich_progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/my_rich_progress_bar.py -------------------------------------------------------------------------------- /models/utils/my_save_config_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/my_save_config_callback.py -------------------------------------------------------------------------------- /models/utils/shared_cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/shared_cli.py -------------------------------------------------------------------------------- /models/utils/snr_sch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/models/utils/snr_sch.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ahmadikalkhorani/CrossNet/HEAD/requirements.txt --------------------------------------------------------------------------------