Pytorch 计算模型参数量
转自:https://blog.csdn.net/jdzwanghao/article/details/84196239
################
### 模型定义
# -------------
class MyModel(nn.Module):
def __init__(self, feat_dim): # input the dim of output fea-ma...
转载
2019-05-11 11:44:15 ·
6312 阅读 ·
0 评论