ModelScope这个bin文件可以用pytorch读取使用吗？
ModelScope这个bin文件可以用pytorch读取使用吗？
展开
收起
1
条回答
写回答
-
ModelScope 的 bin 文件是 TorchScript 格式的模型文件，可以用 torch.jit.load 方法加载。
以下是一个使用 torch.jit.load 加载 TorchScript 格式模型文件的示例代码：
import torch
model = torch.jit.load("model.bin")
input = torch.rand(1, 3, 224, 224)
output = model(input)
更多关于 TorchScript 格式模型文件的使用，请参考 官方文档。2023-09-25 17:34:04赞同 展开评论
相关问答
相关文章
热门讨论
热门文章
展开全部
展开全部
还有其他疑问?
咨询AI助理