편집
1,408
번
바뀜
새 문서: == 작성 == 모델은 class로 구현한다.<syntaxhighlight lang="python"> class Net(nn.Module): def __init__(self) super(Net, self).__init__() 매개변수 정...
== 작성 ==
모델은 class로 구현한다.<syntaxhighlight lang="python">
class Net(nn.Module):
def __init__(self)
super(Net, self).__init__()
매개변수 정의.
</syntaxhighlight>재미있는 방식이다.
<code>model = Net().to(DEVICE)</code> 로 해당 디바이스에 할당하는 것까지가 모델만들기이다.
[[분류:PyTorch]]
모델은 class로 구현한다.<syntaxhighlight lang="python">
class Net(nn.Module):
def __init__(self)
super(Net, self).__init__()
매개변수 정의.
</syntaxhighlight>재미있는 방식이다.
<code>model = Net().to(DEVICE)</code> 로 해당 디바이스에 할당하는 것까지가 모델만들기이다.
[[분류:PyTorch]]