├── Figs ├── FigCompare.jpg └── SturctureFig.jpg ├── LICENSE ├── README.md ├── configs ├── __init__.py ├── callbacks │ ├── default.yaml │ ├── early_stopping.yaml │ ├── model_checkpoint.yaml │ ├── model_summary.yaml │ ├── new_bar.yaml │ ├── none.yaml │ └── rich_progress_bar.yaml ├── data │ ├── CogCap.yaml │ └── mnist.yaml ├── debug │ ├── default.yaml │ ├── fdr.yaml │ ├── limit.yaml │ ├── overfit.yaml │ └── profiler.yaml ├── eval.yaml ├── evalSelf.yaml ├── experiment │ ├── brainencoder_all.yaml │ ├── brainencoder_single.yaml │ └── example.yaml ├── extras │ └── default.yaml ├── hparams_search │ ├── Cogcap.yaml │ └── mnist_optuna.yaml ├── hydra │ ├── default.yaml │ └── seriousrun.yaml ├── logger │ ├── aim.yaml │ ├── comet.yaml │ ├── csv.yaml │ ├── many_loggers.yaml │ ├── mlflow.yaml │ ├── neptune.yaml │ ├── tensorboard.yaml │ └── wandb.yaml ├── model │ ├── Cogcap.yaml │ └── mnist.yaml ├── paths │ ├── default.yaml │ └── pathsSelf.yaml ├── train.yaml └── trainer │ ├── cpu.yaml │ ├── ddp.yaml │ ├── ddp_sim.yaml │ ├── default.yaml │ ├── gpu.yaml │ └── mps.yaml ├── environment.yaml ├── requirements.txt ├── scripts └── schedule.sh ├── setup.py ├── src ├── Scripts │ ├── __init__.py │ ├── generation │ │ ├── __init__.py │ │ └── multiadapter │ │ │ └── main.py │ └── train_align │ │ ├── custompipe.py │ │ ├── diffusion_prior.py │ │ └── main.py ├── __init__.py ├── data │ ├── THINGSEEG_datamodule.py │ ├── __init__.py │ └── components │ │ ├── THINGSEEG_utils.py │ │ ├── __init__.py │ │ └── utils │ │ ├── BLIP2.py │ │ ├── __init__.py │ │ ├── clip_embedding │ │ ├── main.py │ │ ├── main_mid.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── factory.py │ │ │ └── model.py │ │ └── depthanything │ │ ├── __init__.py │ │ ├── blocks.py │ │ ├── dpt.py │ │ ├── main.py │ │ └── transform.py ├── eval.py ├── models │ ├── Cogcap_allmodality_module.py │ ├── Cogcap_single_module.py │ ├── __init__.py │ ├── components │ │ ├── Cogcap │ │ │ ├── Cogcap.py │ │ │ ├── Cogcap_eval.py │ │ │ ├── __init__.py │ │ │ ├── cliploss.py │ │ │ └── loss.py │ │ ├── __init__.py │ │ ├── simple_dense_net.py │ │ └── utils.py │ └── mnist_module.py ├── train.py └── utils │ ├── __init__.py │ ├── instantiators.py │ ├── logging_utils.py │ ├── pylogger.py │ ├── rich_utils.py │ └── utils.py └── tests ├── __init__.py ├── conftest.py ├── helpers ├── __init__.py ├── package_available.py ├── run_if.py └── run_sh_command.py ├── test_configs.py ├── test_datamodules.py ├── test_eval.py ├── test_sweeps.py └── test_train.py /Figs/FigCompare.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/Figs/FigCompare.jpg -------------------------------------------------------------------------------- /Figs/SturctureFig.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/Figs/SturctureFig.jpg -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/__init__.py -------------------------------------------------------------------------------- /configs/callbacks/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/callbacks/default.yaml -------------------------------------------------------------------------------- /configs/callbacks/early_stopping.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/callbacks/early_stopping.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_checkpoint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/callbacks/model_checkpoint.yaml -------------------------------------------------------------------------------- /configs/callbacks/model_summary.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/callbacks/model_summary.yaml -------------------------------------------------------------------------------- /configs/callbacks/new_bar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/callbacks/new_bar.yaml -------------------------------------------------------------------------------- /configs/callbacks/none.yaml: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/callbacks/rich_progress_bar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/callbacks/rich_progress_bar.yaml -------------------------------------------------------------------------------- /configs/data/CogCap.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/data/CogCap.yaml -------------------------------------------------------------------------------- /configs/data/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/data/mnist.yaml -------------------------------------------------------------------------------- /configs/debug/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/debug/default.yaml -------------------------------------------------------------------------------- /configs/debug/fdr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/debug/fdr.yaml -------------------------------------------------------------------------------- /configs/debug/limit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/debug/limit.yaml -------------------------------------------------------------------------------- /configs/debug/overfit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/debug/overfit.yaml -------------------------------------------------------------------------------- /configs/debug/profiler.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/debug/profiler.yaml -------------------------------------------------------------------------------- /configs/eval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/eval.yaml -------------------------------------------------------------------------------- /configs/evalSelf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/evalSelf.yaml -------------------------------------------------------------------------------- /configs/experiment/brainencoder_all.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/experiment/brainencoder_all.yaml -------------------------------------------------------------------------------- /configs/experiment/brainencoder_single.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/experiment/brainencoder_single.yaml -------------------------------------------------------------------------------- /configs/experiment/example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/experiment/example.yaml -------------------------------------------------------------------------------- /configs/extras/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/extras/default.yaml -------------------------------------------------------------------------------- /configs/hparams_search/Cogcap.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/hparams_search/Cogcap.yaml -------------------------------------------------------------------------------- /configs/hparams_search/mnist_optuna.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/hparams_search/mnist_optuna.yaml -------------------------------------------------------------------------------- /configs/hydra/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/hydra/default.yaml -------------------------------------------------------------------------------- /configs/hydra/seriousrun.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/hydra/seriousrun.yaml -------------------------------------------------------------------------------- /configs/logger/aim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/logger/aim.yaml -------------------------------------------------------------------------------- /configs/logger/comet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/logger/comet.yaml -------------------------------------------------------------------------------- /configs/logger/csv.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/logger/csv.yaml -------------------------------------------------------------------------------- /configs/logger/many_loggers.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/logger/many_loggers.yaml -------------------------------------------------------------------------------- /configs/logger/mlflow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/logger/mlflow.yaml -------------------------------------------------------------------------------- /configs/logger/neptune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/logger/neptune.yaml -------------------------------------------------------------------------------- /configs/logger/tensorboard.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/logger/tensorboard.yaml -------------------------------------------------------------------------------- /configs/logger/wandb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/logger/wandb.yaml -------------------------------------------------------------------------------- /configs/model/Cogcap.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/model/Cogcap.yaml -------------------------------------------------------------------------------- /configs/model/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/model/mnist.yaml -------------------------------------------------------------------------------- /configs/paths/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/paths/default.yaml -------------------------------------------------------------------------------- /configs/paths/pathsSelf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/paths/pathsSelf.yaml -------------------------------------------------------------------------------- /configs/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/train.yaml -------------------------------------------------------------------------------- /configs/trainer/cpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/trainer/cpu.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/trainer/ddp.yaml -------------------------------------------------------------------------------- /configs/trainer/ddp_sim.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/trainer/ddp_sim.yaml -------------------------------------------------------------------------------- /configs/trainer/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/trainer/default.yaml -------------------------------------------------------------------------------- /configs/trainer/gpu.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/trainer/gpu.yaml -------------------------------------------------------------------------------- /configs/trainer/mps.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/configs/trainer/mps.yaml -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/environment.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/schedule.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/scripts/schedule.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/setup.py -------------------------------------------------------------------------------- /src/Scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/Scripts/generation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/Scripts/generation/multiadapter/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/Scripts/generation/multiadapter/main.py -------------------------------------------------------------------------------- /src/Scripts/train_align/custompipe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/Scripts/train_align/custompipe.py -------------------------------------------------------------------------------- /src/Scripts/train_align/diffusion_prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/Scripts/train_align/diffusion_prior.py -------------------------------------------------------------------------------- /src/Scripts/train_align/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/Scripts/train_align/main.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/THINGSEEG_datamodule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/THINGSEEG_datamodule.py -------------------------------------------------------------------------------- /src/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/components/THINGSEEG_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/THINGSEEG_utils.py -------------------------------------------------------------------------------- /src/data/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/components/utils/BLIP2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/BLIP2.py -------------------------------------------------------------------------------- /src/data/components/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/components/utils/clip_embedding/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/clip_embedding/main.py -------------------------------------------------------------------------------- /src/data/components/utils/clip_embedding/main_mid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/clip_embedding/main_mid.py -------------------------------------------------------------------------------- /src/data/components/utils/clip_embedding/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/clip_embedding/utils/__init__.py -------------------------------------------------------------------------------- /src/data/components/utils/clip_embedding/utils/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/clip_embedding/utils/factory.py -------------------------------------------------------------------------------- /src/data/components/utils/clip_embedding/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/clip_embedding/utils/model.py -------------------------------------------------------------------------------- /src/data/components/utils/depthanything/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data/components/utils/depthanything/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/depthanything/blocks.py -------------------------------------------------------------------------------- /src/data/components/utils/depthanything/dpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/depthanything/dpt.py -------------------------------------------------------------------------------- /src/data/components/utils/depthanything/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/depthanything/main.py -------------------------------------------------------------------------------- /src/data/components/utils/depthanything/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/data/components/utils/depthanything/transform.py -------------------------------------------------------------------------------- /src/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/eval.py -------------------------------------------------------------------------------- /src/models/Cogcap_allmodality_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/models/Cogcap_allmodality_module.py -------------------------------------------------------------------------------- /src/models/Cogcap_single_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/models/Cogcap_single_module.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/Cogcap/Cogcap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/models/components/Cogcap/Cogcap.py -------------------------------------------------------------------------------- /src/models/components/Cogcap/Cogcap_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/models/components/Cogcap/Cogcap_eval.py -------------------------------------------------------------------------------- /src/models/components/Cogcap/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/Cogcap/cliploss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/models/components/Cogcap/cliploss.py -------------------------------------------------------------------------------- /src/models/components/Cogcap/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/models/components/Cogcap/loss.py -------------------------------------------------------------------------------- /src/models/components/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/components/simple_dense_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/models/components/simple_dense_net.py -------------------------------------------------------------------------------- /src/models/components/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/models/components/utils.py -------------------------------------------------------------------------------- /src/models/mnist_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/models/mnist_module.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/instantiators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/utils/instantiators.py -------------------------------------------------------------------------------- /src/utils/logging_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/utils/logging_utils.py -------------------------------------------------------------------------------- /src/utils/pylogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/utils/pylogger.py -------------------------------------------------------------------------------- /src/utils/rich_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/utils/rich_utils.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/src/utils/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/conftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/tests/conftest.py -------------------------------------------------------------------------------- /tests/helpers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/helpers/package_available.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/tests/helpers/package_available.py -------------------------------------------------------------------------------- /tests/helpers/run_if.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/tests/helpers/run_if.py -------------------------------------------------------------------------------- /tests/helpers/run_sh_command.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/tests/helpers/run_sh_command.py -------------------------------------------------------------------------------- /tests/test_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/tests/test_configs.py -------------------------------------------------------------------------------- /tests/test_datamodules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/tests/test_datamodules.py -------------------------------------------------------------------------------- /tests/test_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/tests/test_eval.py -------------------------------------------------------------------------------- /tests/test_sweeps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/tests/test_sweeps.py -------------------------------------------------------------------------------- /tests/test_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XiaoZhangYES/CognitionCapturer/HEAD/tests/test_train.py --------------------------------------------------------------------------------