摘要: 这里是ubuntu系统,其他系统我忘了 ls -l 出来有没有total字符了。 ls -l | grep -v total | wc -l 阅读全文
posted @ 2023-10-27 21:13 Dsp Tian 阅读(15) 评论(0) 推荐(0) 编辑
摘要: 统计模型参数量,方便判断不同模型大小: import torch import torch.nn as nn class AlexNet(nn.Module): def __init__(self): super(AlexNet, self).__init__() self.conv1 = nn.C 阅读全文
posted @ 2023-10-27 21:01 Dsp Tian 阅读(70) 评论(0) 推荐(0) 编辑