2024年1月15日
pytorch调用训练的模型
摘要： import torch # 加载模型参数 model_path = 'model.pth' model_state_dict = torch.load(model_path) # 创建模型实例 input_size = 10 # 输入层大小，根据实际情况调整 hidden_size1 = 32 # 阅读全文
posted @ 2024-01-15 21:11 西北逍遥 阅读(108) 评论(0) 推荐(0) 编辑