求教如何训练 StyleGan 的 classifier（分类器）

最近领了一个人脸编辑的任务，大概就是输入一张人脸然后调整为微笑、大眼睛，不一样的是我们要的是“痛苦”的表情，试了 StyleGan 相关的编辑，可以跑通官方提供的模型，但是 CeleA 没有“痛苦“属性，需要自己动手，训练以恶类似 celebahq-classifier-00-male.pkl 的模型。

看了资料是说要训练一个分类器，我尝试用 ResNet50 跑了一个，但是无法导入。看了 StyleGan 论文才的分类器结构和 Discriminator 一样。大概理解可以用代码里的 D_basic 跑一个模型，然后使用？瞎折腾几天了，感觉方向对了，道路很曲折 /sad 。

求大佬指点。刚接触机器学习，表示了解不多。