coqui-tts/TTS/utils
Agrin Hilmkil ced4cfdbbf Allow saving / loading checkpoints from cloud paths (#683)
* Allow saving / loading checkpoints from cloud paths

Allows saving and loading checkpoints directly from cloud paths like
Amazon S3 (s3://) and Google Cloud Storage (gs://) by using fsspec.

Note: The user will have to install the relevant dependency for each
protocol. Otherwise fsspec will fail and specify which dependency is
missing.

* Append suffix _fsspec to save/load function names

* Add a lower bound to the fsspec dependency

Skips the 0 major version.

* Add missing changes from refactor

* Use fsspec for remaining artifacts

* Add test case with path requiring fsspec

* Avoid writing logs to file unless output_path is local

* Document the possibility of using paths supported by fsspec

* Fix style and lint

* Add missing lint fixes

* Add type annotations to new functions

* Use Coqpit method for converting config to dict

* Fix type annotation in semi-new function

* Add return type for load_fsspec

* Fix bug where fs not always created

* Restore the experiment removal functionality
2021-08-09 18:02:36 +00:00
..
logging Fixup `utils` for the trainer 2021-06-28 17:03:47 +02:00
__init__.py rename the project to old TTS 2020-09-09 12:27:23 +02:00
audio.py Update glowtts docstrings and docs 2021-06-30 14:30:55 +02:00
callbacks.py Implement unified trainer 2021-06-28 17:03:19 +02:00
distribute.py Implement unified trainer 2021-06-28 17:03:19 +02:00
generic_utils.py Allow saving / loading checkpoints from cloud paths (#683) 2021-08-09 18:02:36 +00:00
io.py Allow saving / loading checkpoints from cloud paths (#683) 2021-08-09 18:02:36 +00:00
manage.py Fix `ModelManager` model download 2021-07-02 10:47:05 +02:00
radam.py Fixup `utils` for the trainer 2021-06-28 17:03:47 +02:00
synthesizer.py Let `Synthesizer` use the speaker manager out of the model 2021-07-02 10:47:55 +02:00
trainer_utils.py Raise an error when multiple GPUs are in use 2021-07-04 11:25:49 +02:00
training.py Implement unified trainer 2021-06-28 17:03:19 +02:00