Dsp Tian

摘要： 这里是ubuntu系统，其他系统我忘了 ls -l 出来有没有total字符了。 ls -l | grep -v total | wc -l 阅读全文

摘要： 统计模型参数量，方便判断不同模型大小： import torch import torch.nn as nn class AlexNet(nn.Module): def __init__(self): super(AlexNet, self).__init__() self.conv1 = nn.C 阅读全文