search

essunny310 / FreestyleNet

[CVPR 2023 Highlight] Freestyle Layout-to-Image Synthesis
https://essunny310.github.io/FreestyleNet/
MIT License
149 stars 3 forks source link

What is the VRAM requirement? #2

Closed Njasa2k closed 1 year ago

essunny310 commented 1 year ago

Hi, we managed to train FreestyleNet on a single RTX 3090 GPU (24G VRAM) with batch size = 1. For the inference, 10G VRAM is fine in our case.