siren icon indicating copy to clipboard operation
siren copied to clipboard

Training videos. Missing test_video.py?

Open RKelln opened this issue 4 years ago • 0 comments

Trained up a a custom video but then realized there was no equivalent to test_audio.py. Is that just an oversight, as there is an example video on the project page?

Also note that there seems to be missing support for checkpoint_path in train_video.py, something like this can be added:

if opt.checkpoint_path is not None:
    model.load_state_dict(torch.load(opt.checkpoint_path))

In my training the loss started increasing after hitting a minimum around 20000. Is that what you observed?

RKelln avatar Mar 18 '21 13:03 RKelln