CyberAgentAILab / layout-dm

LayoutDM: Discrete Diffusion Model for Controllable Layout Generation [Inoue+, CVPR2023]
https://cyberagentailab.github.io/layout-dm
Apache License 2.0
225 stars 24 forks source link

Hardware requirement #40

Closed hsinwu13 closed 5 months ago

hsinwu13 commented 6 months ago

image

I would like to ask the hardware requirement for the training? Thank you.

naoto0804 commented 6 months ago

Hi, thank you for your interest in our project! I was working on T4 (16GB), so it should work. Do you work on a custom dataset?

naoto0804 commented 5 months ago

Closed due to inactivity. Feel free to re-open if you have more questions!