vsitzmann / siren

Official implementation of "Implicit Neural Representations with Periodic Activation Functions"
MIT License
1.72k stars 247 forks source link

Training videos. Missing test_video.py? #42

Open RKelln opened 3 years ago

RKelln commented 3 years ago

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?