Code to train Encodec

[nicholasjsanders]

There is a train.py script within the encoder directory, however when attempting to run python srcs.encodec.train.py --help or python encodec.train.py --help there are circular importation errors with from . import modules as m within encodec/model.py. Also, within srcs/modules/init.py there is a line from .transformer_discrete import Transformer but there is no transformer_discrete file. Is there a different way to run this code successfully or are there some naming discrepancies that need to be updated?

[haiciyang]

Thank you for your question. Sorry for the version discrepancy. On the current version, you can comment this line - from .transformer_discrete import Transformer; It's not used anywhere.

For training discrete auto-encoders - try this script, python -m srcs.train --lr 0.00005 --seq_len_p_sec 2.4 --rep_dims 128 --n_residual_layers 1 --enc_ratios 8 5 4 2 --quantization --bandwidth 1.5 --n_filters 32 --lstm 2 --model_type unet --seq_length 1200 --data_folder_path [DATA FOLDER] --use_disc --disc_freq 5 Thanks