Giters

jixinya / EAMM

Code for paper 'EAMM: One-Shot Emotional Talking Face via Audio-Based Emotion-Aware Motion Model'

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Errors in train.py

neeek2303 opened this issue · comments

commented

Hello, I found a few errors in train.py:

  1. 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

  2. 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