复现MTCNN网络模型,对输出的class、box分别计算loss
【操作步骤&问题现象】
1、此处是否可以使用WithLossCell&TrainOneStepCell方法呢?如果可以应该怎样操作呢?
2、如果不可以用,是否有其他方法可以实现,能否提供参考样例?
原因:如图所示,num_epochs参数之前设置的是1,改成max_epoch就好了
复现MTCNN网络模型,对输出的class、box分别计算loss
【操作步骤&问题现象】
1、此处是否可以使用WithLossCell&TrainOneStepCell方法呢?如果可以应该怎样操作呢?
2、如果不可以用,是否有其他方法可以实现,能否提供参考样例?
原因:如图所示,num_epochs参数之前设置的是1,改成max_epoch就好了