Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Eric_lai 36692044aa | 1 year ago | |
---|---|---|
.. | ||
__init__.py | 1 year ago | |
comparison_by_inference .py | 1 year ago | |
convert_tensor.py | 1 year ago | |
pth2ckpt.py | 1 year ago | |
readme.md | 1 year ago |
pth2ckpt.py
文件中的pth2ckpt
接口可以将Torch的权重文件转换为MindSpore的权重文件。使用方法如下:
from pth2ckpt import pth2ckpt
path = './xx.pth'
pth2ckpt(path)
# "convert ckpt finish."
convert_tensor.py
文件中的convert_to_ms_tensor
方法是过渡阶段的Tensor转换接口。该接口用于将torch.utils.data.DataLoader
迭代输出的数据转换成为MindSpore框架张量数据。
在过渡阶段数据预处理和数据集相关接口和过程均使用torch原生接口,导入的包使用torch以及torchvision。为了避免导入包之间冲突请务必保证文件结构如下:
folder/
└── dataset_and_dataloader.py
└── model_and_train.py
其中dataset_and_dataloader.py
文件中编写数据预处理操作以及数据集处理,此时可以导入torch包进行处理。
from torch.utils.data.dataset import Dataset
from torch.utils.data.dataloader import DataLoader
dataset = make_dataset ...
train_dataloader = DataLoader(dataset)...
model_and_train.py
文件中编写模型文件和训练过程,此时可以导入MSAdapter包进行处理。
该接口对迭代数据进行转换:
from ms_adapter.pytorch.utils.data import convert_to_ms_tensor
from dataset_and_dataloader import train_dataloader, test_dataloader
for x, y in train_dataloader:
x = convert_to_ms_tensor(x)
y = convert_to_ms_tensor(y)
model_train... or model_inference...
No Description
Jupyter Notebook C++ Python Text C other
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》