search

KichangKim / DeepDanbooru

AI based multi-label girl image classification system, implemented by using TensorFlow.
MIT License
2.65k stars 260 forks source link

Think of label classification as Multi-Task Learning #65

Closed OysterQAQ closed 2 years ago

OysterQAQ commented 2 years ago

Have you tried to split the system label and text label to form a multi-output model with different tasks? How many epochs did you end up training before the model converged?

KichangKim commented 2 years ago

No, I didn't try to split model. It always has combined model. All of released models are trained with 30 epochs. 1 epoch has about 1.7m images.