Loading

2022年5月29日
pytorch模型定义常用函数以及resnet模型修改案例
摘要： 模型定义常用函数 利用nn.Parameter()设计新的层 import torch from torch import nn class MyLinear(nn.Module): def __init__(self, in_features, out_features): super().__i 阅读全文
posted @ 2022-05-29 16:16 MapleTx 阅读(700) 评论(0) 推荐(0) 编辑