B导B导，在用你的pytorch图像分类时候，我有个三分类任务，数据量很小，大概各200例图像，以不冻结形式train，我想把epoch设置小一点，怕过拟合，该怎么弄呀，一直warning让我把epoch设置300以上。第二个问题，我还有各自己搭建的模型，数据是不平衡分布的，我是想以BBN那个框架为基础，用B导您的代码训练该怎么改呢@