@@ -21,7 +21,7 @@ This example of TensorFlow.js runs simultaneously in two different environments:
21
21
- Demonstration of generation in the browser. The demo webpage will load
22
22
the checkpoints saved from the training process and use it to generate
23
23
fake MNIST images in the browser.
24
-
24
+
25
25
## How to use this example
26
26
27
27
This example can be used in two ways:
@@ -32,7 +32,7 @@ This example can be used in two ways:
32
32
the web.
33
33
34
34
For approach 1, you can start the training by:
35
-
35
+
36
36
``` sh
37
37
yarn
38
38
yarn train
@@ -71,6 +71,9 @@ advantanges:
71
71
72
72
![ MNIST ACGAN Training: TensorBoard Example] ( ./mnist-acgan-tensorboard-example.png )
73
73
74
+ Detailed loss profiles are hosted and viewable at this
75
+ [ TensorBoard.dev link] ( https://tensorboard.dev/experiment/iBcGONlbQbmVyNd8H6unJg/#scalars ) .
76
+
74
77
To do this in this example, add the flag ` --logDir ` to the ` yarn train `
75
78
command, followed by the directory to which you want the logs to
76
79
be written, e.g.,
@@ -93,9 +96,9 @@ navigate to the URL to view the loss curves in the Scalar dashboard of
93
96
TensorBoard.
94
97
95
98
### Running Generator demo in the Browser
96
-
99
+
97
100
To start the demo in the browser, do in a separate terminal:
98
-
101
+
99
102
``` sh
100
103
yarn
101
104
yarn watch
@@ -109,13 +112,13 @@ been started), the user may still click the "Load Hosted Model" button
109
112
to load a remotely-hosted generator.
110
113
111
114
### Training the model on CUDA GPUs using tfjs-node-gpu
112
-
115
+
113
116
It is recommended to use tfjs-node-gpu to train the model on a CUDA-enabled GPU,
114
117
as the convolution heavy operations run several times faster a GPU than on the
115
118
CPU with tfjs-node.
116
119
117
120
By default, the [ training script] ( ./gan.js ) runs on the CPU using tfjs-node. To
118
- run it on the GPU, repace the line
121
+ run it on the GPU, repace the line
119
122
120
123
``` js
121
124
require (' @tensorflow/tfjs-node' );
0 commit comments