存储库中有一个预训练模型,其文件类型为.pyth.我在网上搜索了一下这个文件类型,以及哪种语言能够阅读这个文件,但我什么也没找到.由于我正在使用Pytork,是否可以在Pytork中读取此类文件?此外,通常情况下,如何读取和生成该信息?
更清楚的是,在repository of the TimeSformer模型中,预训练模型属于这种文件类型,例如,您可以在该存储库中找到以下命令:
import torch
from timesformer.models.vit import TimeSformer
model = TimeSformer(img_size=224, num_classes=400, num_frames=8, attention_type='divided_space_time', pretrained_model='/path/to/pretrained/model.pyth')
dummy_video = torch.randn(2, 3, 8, 224, 224) # (batch x channels x frames x height x width)
pred = model(dummy_video,) # (2, 400)