2024年6月17日
pytorch使用交叉熵训练模型学习笔记
摘要： python代码： import torch import torch.nn as nn import torch.optim as optim # 定义一个简单的神经网络模型 class SimpleModel(nn.Module): def __init__(self): super(Simpl 阅读全文
posted @ 2024-06-17 17:26 阿坦 阅读(18) 评论(0) 推荐(0) 编辑