CUDA memory for training Morph_UGATIT

[vinduon]

Dear sir,

Could you tell me how many GPU memory used for default training setting in your configs? I only have 12 GB for GPU so which one should I reduce in the file cfgs_s_ugatit_plus.py? Thank you, sir.

[shoutOutYangJie]

@duongquangvinh I will check the training code. I use 4 rtx2080ti with 11GB. You should train successfully.

[vinduon]

@duongquangvinh I will check the training code. I use 4 rtx2080ti with 11GB. You should train successfully.

Oh thank you for your reply sir!
I have another question that can we save checkpoints after a certain iterations/epochs? In UGATIT we can save checkpoint after 100k iterations as the default. Thank you sir.