search

Kibeom-Hong / Domain-Aware-Style-Transfer

Official Implementation of Domain-Aware Universal Style Transfer
Other
102 stars 11 forks source link

About GPU use #8

Open KevinMarkVine opened 3 years ago

KevinMarkVine commented 3 years ago

I trained StyleIndicator and Decoder respectively. However, I found the Volatile GPU-Util is 0%. It seems that the whole network computed in CPU. Have you ever had this situation？

KevinMarkVine commented 3 years ago
1
KevinMarkVine commented 3 years ago

Actually, it appears when training the decoder

Kibeom-Hong commented 2 years ago

Thanks for your interest and sorry for late replying.

Actually I also found that problem sometimes but GPU utilization came back to 100% as soon.

I think if there is some bottleneck in CPU for dataloader, GPU-util will go to 0%.

If there any other opinion, please let me know

Thank you!