无法在Google Colab中加载预先训练的模型

Question

model_save_name = 'mnist_simple_cnn.h5' path = F"/content/gdrive/My Drive/{model_save_name}" torch.save(model.state_dict(), path)

此代码给出错误： `NameError Traceback（最近一次通话最近）

<ipython-input-12-a3a1df387b48> in <module>()
      1 model_save_name = 'mnist_simple_cnn.h5'
      2 path = F"/content/gdrive/My Drive/{model_save_name}"
----> 3 torch.save(model.state_dict(), path)

NameError: name 'torch' is not defined`