Skip to content

Commit 4d4e859

Browse files
Tutorial2: moved functs from utils.py to notebook
1 parent e63a035 commit 4d4e859

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

Tutorial2/Tutorial2.ipynb

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -310,6 +310,9 @@
310310
"metadata": {},
311311
"outputs": [],
312312
"source": [
313+
"import torch.nn as nn\n",
314+
"import torch\n",
315+
"\n",
313316
"#%% Linear layer\n",
314317
"class LinearModel(nn.Module):\n",
315318
" def __init__(self, input_dim, output_dim):\n",

0 commit comments

Comments
 (0)