EAMM
EAMM copied to clipboard
Errors in train.py
Hello, I found a few errors in train.py:
-
Traceback (most recent call last): File "/fsx/nikitadrobyshev/EAMM/run.py", line 121, in
train_part1(config, generator, discriminator, kp_detector, kp_detector_a,audio_feature, opt.checkpoint, opt.audio_checkpoint, log_dir, dataset, test_dataset,opt.device_ids, name) File "/fsx/nikitadrobyshev/EAMM/train.py", line 26, in train_part1 optimizer_generator, optimizer_discriminator, NameError: name 'optimizer_generator' is not defined -
File "/fsx/nikitadrobyshev/EAMM/train.py", line 55, in train_part1 generator_full = TrainPart1Model(kp_detector, kp_detector_a, audio_feature, generator, discriminator, train_params,device_ids) File "/fsx/nikitadrobyshev/EAMM/modules/model.py", line 164, in init super(TrainFullModel, self).init() NameError: name 'TrainFullModel' is not defined