如何确定 google colab 中 .pth 文件的形状

问题描述

我有 abc.pth 文件，它是一个功能文件。我想通过 google colab 确定此文件的形状/尺寸。如何加载并确定其形状？

解决方法

path="/path/to/abc.pth"

导入火炬 a=torch.load(path) a.size()

它会输出张量

torch.Size([1,2048,19,29])