torch.nn.Parameter 에 관해서

Pytorch 에는 Parameter라는 모듈이 있는데, 얘는 레이어가 아니라 말 그대로 파라미터 값만을 가지고 있는 놈이다. class Actor(nn.Module): def __init__(self, num_inputs, num_outputs, continuous=True, shared=False): self.num_inputs = num_inputs self.num_outputs = num_outputs super(Actor, self).__init__() self.fc1 = nn.Linear(num_inputs, hp.hidden) self.fc2 = nn.Linear(hp.hidden, hp.hidden) self.fc3 = nn.Linear(hp.hidden, num_outputs) self...

원문링크 : torch.nn.Parameter 에 관해서