search

lucidrains / magvit2-pytorch

Implementation of MagViT2 Tokenizer in Pytorch
MIT License
565 stars 34 forks source link

Example for Images #3

Closed jpfeil closed 1 year ago

jpfeil commented 1 year 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 1 year 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 1 year ago

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