在烧瓶中加载预先训练的模型

Question

我有一个使用 nn.Module 的项目。训练我制作的模型后，我使用代码保存模型

torch.save(model_basic_cnn.state_dict(), 'CNN.pth')
weights_fname = 'CNN.pth'

我的问题是如何加载我之前保存在烧瓶中的模型。

注意：对于flask，我使用VS Code，对于训练模型使用google colab