File tree Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Expand file tree Collapse file tree 1 file changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -109,10 +109,12 @@ one channel instead of three and 28x28 instead of 32x32:
109
109
return x
110
110
111
111
112
- net = Net()
112
+ net = Net()
113
113
114
114
We'll define the same ``optimizer `` and ``criterion `` from before:
115
115
116
+ .. code :: python
117
+
116
118
criterion = nn.CrossEntropyLoss()
117
119
optimizer = optim.SGD(net.parameters(), lr = 0.001 , momentum = 0.9 )
118
120
@@ -136,7 +138,7 @@ folder.
136
138
~~~~~~~~~~~~~~~~~~~~~~~~~
137
139
138
140
Now let's write an image to our TensorBoard - specifically, a grid -
139
- using `make_grid <https://pytorch.org/docs/stable/torchvision/utils.html#torchvision.utils.make_grid >`__
141
+ using `make_grid <https://pytorch.org/docs/stable/torchvision/utils.html#torchvision.utils.make_grid >`__.
140
142
141
143
.. code :: python
142
144
@@ -157,9 +159,9 @@ Now running
157
159
158
160
::
159
161
160
- ` tensorboard --logdir=runs`
162
+ tensorboard --logdir=runs
161
163
162
- from the command line and then navigating to `https://localhost:6006 `
164
+ from the command line and then navigating to `https://localhost:6006 < https://localhost:6006 >`_
163
165
should show the following.
164
166
165
167
.. image :: ../../_static/img/tensorboard_first_view.png
You can’t perform that action at this time.
0 commit comments