PaddleClas 如何实现模型在train 以及 infer 的时候使用不同分支的forword

[violetweir]

您好麻烦询问一下
RepVgg在训练的时候会使用3x3 1x1 de1Repvgg 模块 在模型导出以及预测的时候可以将3x3 和1x1 的block合成一个block
我也想实现在训练和预测的时候在forward阶段进行不同的操作 paddleclas以及paddle应该如何做才可以呀

[changdazhou]

将forward写成两部分，根据模型的训练状态还是预测状态调用各自部分就好，例如：

def forward(self,state,input):
    if state is 'infer':
        forward_infer(input)
    else:
        forward_train_eval(input)