img_size 512

Author: Thomas Capelle

train_uvit.py

@@ -25,7 +25,7 @@
     sampler_steps=333, # number of sampler steps on the diffusion process
     seed = 42, # random seed
     batch_size = 6, # batch size
-    img_size = 128, # image size
+    img_size = 512, # image size
     device = "cuda", # device
     num_workers=8, # number of workers for dataloader
     num_frames=4, # number of frames to use as input