Related to: Machine Learning

torch.nn.Parameter

  • Tensor를 상속받는 Class
  • nn.Moudule 내에 Attribute가 될 때, required_grad=True로 지정되어 자동으로 학습대상이 됨

참조

Week 2