512x512 training configuration

[why529913]

I want to know the specific configuration of 512x512 size training, --num_bucket_per_side should be 512, 512?

[KaiChen1998]

The num_bucket_per_side for nuImages is 512x512, while 256x256 for COCO-Stuff. Check the generation configs in our pre-trained checkpoints for reference here.

Other configurations are maintained similarly to the 256x256 experiments. The only thing to change is to linearly scale the learning rate with respect to the batch size.