net = slowfastnet.resnet50(class_num=params['num_classes'])
#加载预训练模型
net_dict = torch.load(
"D:\\SlowFastNetworks-master\\UCF101\\2020-04-20-18-04-06\\clip_len_64frame_sample_rate_1_checkpoint_95.pth.tar")</
加载预训练模型解密
最新推荐文章于 2024-08-21 00:32:13 发布
本文介绍了如何在PyTorch中加载预训练模型的步骤,包括初始化网络、使用`torch.load()`加载权重、通过`load_state_dict()`方法将权重加载到网络中。强调了`load_state_dict()`与`load`的区别。
摘要由CSDN通过智能技术生成