is there any GPU acceleration?

KichangKim / DeepDanbooru

AI based multi-label girl image classification system, implemented by using TensorFlow.

MIT License

2.58k stars 258 forks source link

is there any GPU acceleration? #91

Closed delislej closed 1 year ago

delislej commented 1 year ago

hi there, i have successfully started training, but it says it will take over a month with a dataset of 280k images.

i have a 3090 with 24gb of vram, you mentioned that you trained it on a 3070, so i'm wondering if it is just the size of my data set or if i have improperly configured something. thanks!

delislej commented 1 year ago

please close this issue, it was an installation issue on my end.