├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── configs └── lorenz.yaml ├── datasets └── lorenz_dataset.h5 ├── example_scripts ├── ray_test.py ├── run_pbt.py ├── run_random_search.py └── train_lfads.py ├── lfads-tf2 ├── lfads_tf2 │ ├── __init__.py │ ├── defaults.py │ ├── initializers.py │ ├── layers.py │ ├── logging_conf.yaml │ ├── models.py │ ├── regularizers.py │ ├── tuples.py │ └── utils.py └── setup.py ├── ray_cluster_template.yaml └── tune-tf2 ├── setup.py └── tune_tf2 ├── __init__.py ├── defaults.py ├── models.py ├── pbt ├── __init__.py ├── exploiters.py ├── explorers.py ├── hps.py ├── schedulers.py ├── trial_executor.py └── utils.py ├── randsearch.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/README.md -------------------------------------------------------------------------------- /configs/lorenz.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/configs/lorenz.yaml -------------------------------------------------------------------------------- /datasets/lorenz_dataset.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/datasets/lorenz_dataset.h5 -------------------------------------------------------------------------------- /example_scripts/ray_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/example_scripts/ray_test.py -------------------------------------------------------------------------------- /example_scripts/run_pbt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/example_scripts/run_pbt.py -------------------------------------------------------------------------------- /example_scripts/run_random_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/example_scripts/run_random_search.py -------------------------------------------------------------------------------- /example_scripts/train_lfads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/example_scripts/train_lfads.py -------------------------------------------------------------------------------- /lfads-tf2/lfads_tf2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lfads-tf2/lfads_tf2/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/lfads-tf2/lfads_tf2/defaults.py -------------------------------------------------------------------------------- /lfads-tf2/lfads_tf2/initializers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/lfads-tf2/lfads_tf2/initializers.py -------------------------------------------------------------------------------- /lfads-tf2/lfads_tf2/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/lfads-tf2/lfads_tf2/layers.py -------------------------------------------------------------------------------- /lfads-tf2/lfads_tf2/logging_conf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/lfads-tf2/lfads_tf2/logging_conf.yaml -------------------------------------------------------------------------------- /lfads-tf2/lfads_tf2/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/lfads-tf2/lfads_tf2/models.py -------------------------------------------------------------------------------- /lfads-tf2/lfads_tf2/regularizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/lfads-tf2/lfads_tf2/regularizers.py -------------------------------------------------------------------------------- /lfads-tf2/lfads_tf2/tuples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/lfads-tf2/lfads_tf2/tuples.py -------------------------------------------------------------------------------- /lfads-tf2/lfads_tf2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/lfads-tf2/lfads_tf2/utils.py -------------------------------------------------------------------------------- /lfads-tf2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/lfads-tf2/setup.py -------------------------------------------------------------------------------- /ray_cluster_template.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/ray_cluster_template.yaml -------------------------------------------------------------------------------- /tune-tf2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/setup.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/defaults.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/models.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/pbt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/pbt/exploiters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/pbt/exploiters.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/pbt/explorers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/pbt/explorers.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/pbt/hps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/pbt/hps.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/pbt/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/pbt/schedulers.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/pbt/trial_executor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/pbt/trial_executor.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/pbt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/pbt/utils.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/randsearch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/randsearch.py -------------------------------------------------------------------------------- /tune-tf2/tune_tf2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snel-repo/autolfads-tf2/HEAD/tune-tf2/tune_tf2/utils.py --------------------------------------------------------------------------------