Does your model works in only one class of afhq?

[justinday123]

for example, if you train with all of classes in AFHQ, doesn't it works well?