import mindspore as ms
from mindspore import nn, ops

# 定义一个简单的神经网络模型
class SimpleNet(nn.Cell):
    def __init__(self):
        super(SimpleNet, self).__init__()
        self.fc = nn.Dense(10, 2)

    def construct(self, x):
        return self.fc(x)

# 初始化模型和优化器
net = SimpleNet()
optimizer = nn.Adam(net.trainable_params(), learning_rate=0.01)

# 定义损失函数
loss_fn = nn.MSELoss()

# 模拟数据
x = ops.randn((32, 10))
y = ops.randn((32, 2))

# 训练循环
for _ in range(100):
    loss = loss_fn(net(x), y)
    optimizer(loss)
    print(f"Loss: {loss.asnumpy()}")

Model Agent 是昇腾面向开发者推出的端到端 AI 模型自动化适配工具，旨在简化 AI 模型在昇腾平台上的部署和优化过程。通过 Model Agent，开发者可以快速将模型适配到昇腾芯片上，实现高效的推理和训练。

参考来源

• DeepSeek V4去CUDA化实践：国产AI全栈迁移技术解析