这是机器学习的新手-我有一个训练脚本,该训练脚本使用图像作为输入,将没有权重的默认模型训练到分类器中,并且在训练后,它提供带有.hdf5文件的输出。
我知道.hdf5文件将包含模型的信息和权重(https://www.tinymind.com/learn/terms/hdf5)。但是我试图找到一种方法,将输出的.hdf5文件用于使用pytorch进行对象分类的特定脚本。
搜索后如何在pytorch中使用此输出文件格式作为模型的信息还很少。
我需要做什么才能将该输出文件用作在火炬中使用的训练模型?如果有人指出正确的方向开始使用,我将不胜感激