search

lucidrains / magvit2-pytorch

Implementation of MagViT2 Tokenizer in Pytorch
MIT License
538 stars 35 forks source link

Example for Images #3

Closed jpfeil closed 10 months ago

jpfeil commented 10 months ago

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!

lucidrains commented 10 months ago

@jpfeil hey Jacob! thanks for your interest

you can pretrain on images by setting the dataset_type to "images" https://github.com/lucidrains/magvit2-pytorch/commit/ec2696d55f0c4cb1028b258e621ed207873fc7b1

lucidrains commented 10 months ago

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