lucidrains/magvit2-pytorch

Example for Images

jpfeil opened this issue · 2 comments

jpfeil commented

I'm excited to apply this model to image data. I was wondering if you could give a trivial example for how to apply this library to individual images instead of video. Thank you!

@jpfeil hey Jacob! thanks for your interest

you can pretrain on images by setting the dataset_type to "images" ec2696d

🤔 i need to take my dog down to Santa Cruz at some point for a hike, and for the boardwalk of course