coqui-tts/tests/tts_tests
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
..
__init__.py move bash script based tests to python with coqpit 2021-05-11 11:29:17 +02:00
test_align_tts_train.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_feed_forward_layers.py update `sequence_mask` import globally 2021-06-28 17:03:19 +02:00
test_glow_tts.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_glow_tts_train.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_speedy_speech_layers.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_speedy_speech_train.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_tacotron2_d-vectors_train.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_tacotron2_model.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_tacotron2_speaker_emb_train.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_tacotron2_tf_model.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_tacotron2_train.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_tacotron2_train_fsspec_path.py Allow saving / loading checkpoints from cloud paths (#683) 2021-08-09 18:02:36 +00:00
test_tacotron_layers.py add max_decoder_steps argument to tacotron models 2021-06-28 17:03:19 +02:00
test_tacotron_model.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00
test_tacotron_train.py Update tests for the new trainer API 2021-06-28 17:03:19 +02:00