Upload files to "/"
This commit is contained in:
parent
b772fe9e8b
commit
732aceb5ac
10
example.py
Normal file
10
example.py
Normal file
|
@ -0,0 +1,10 @@
|
|||
import torch
|
||||
import torch.nn as nn
|
||||
|
||||
class MyModel(nn.Module):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
self.fc = nn.Linear(10, 2)
|
||||
|
||||
def forward(self, x):
|
||||
return self.fc(x)
|
Loading…
Reference in New Issue
Block a user