search

google-research / magvit

Official JAX implementation of MAGVIT: Masked Generative Video Transformer
https://magvit.cs.cmu.edu
Apache License 2.0
956 stars 42 forks source link

Training 2D VQGAN on UCF 101 #19

Open bridenmj opened 9 months ago

bridenmj commented 9 months ago

Hello, I am attempting to train 2D VQGAN with UCF 101 and do inflation. I call the following function:

python main.py --workdir=../out --config=configs/vqgan2d_ucf101_config.py

I get the following error:

KeyError: 'Unknown dataset (ucf101). Did you import the dataset module explicitly?'

After investigating scenic, I see there is no option for ufc101 in _IMPORT_TABLE. Is there a way to work around this?

shinshiner commented 7 months ago

have you solved this?