摘要:
这里是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 阅读全文