2024年8月7日
pytorch深度学习分类代码简单示例
摘要： train.py代码如下 import torch import torch.nn as nn import torch.optim as optim model_save_path = "my_model.pth" # 定义简单的线性神经网络模型 class MyModel(nn.Module): 阅读全文
posted @ 2024-08-07 10:31 阿坦 阅读(21) 评论(0) 推荐(0) 编辑