menyifang / DCT-Net

Official implementation of "DCT-Net: Domain-Calibrated Translation for Portrait Stylization", SIGGRAPH 2022 (TOG); Multi-style cartoonization
Apache License 2.0
734 stars 74 forks source link

How to train on GPU? #65

Closed h3clikejava closed 7 months ago

h3clikejava commented 7 months ago

I am training with NVIDIA 4090 single card using 2/330000 [01:13<3334:32:11, 36.38s/it]