self.modules() 和 self.children()的区别
代码:
import torch
import torch.nn as nn
input_nc = 3
hide1_nc = 6
hide2_nc = 6
output_nc = 3
class Net(nn.Module):
def __init__(self, input_nc, hide1_nc, hide2_nc, output_nc):
super()._...
原创
2019-08-08 12:37:34 ·
1852 阅读 ·
0 评论