jit保存静态模型后，批量预测报错
收藏
在搭建模型时，forward方法中有根据x.shape得到batch_size，在保存静态模型时，shape设置为[None, 3, 224, 224]，如下图所示
在用静态模型预测时，得到的batch_size 为-1，导致程序出错，如下图所示，求大神指教如何修改
0
收藏
请登录后评论
