Skip to content

Commit 01f0f08

Browse files
committed
Changed default parameter.
1 parent 494caa8 commit 01f0f08

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

neural_structured_learning/research/gam/experiments/run_train_mnist.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@
194194
flags.DEFINE_integer(
195195
'batch_size_cls', 512, 'Batch size for classification model.')
196196
flags.DEFINE_float(
197-
'gradient_clip', None,
197+
'gradient_clip', 10,
198198
'The gradient clipping global norm value. If None, no clipping is done.')
199199
flags.DEFINE_integer(
200200
'logging_step_cls', 200,
@@ -278,10 +278,10 @@
278278
'optimizer', 'adam',
279279
'Which optimizer to use. Valid options are `adam`, `amsgrad`.')
280280
flags.DEFINE_bool(
281-
'load_from_checkpoint', False,
282-
'Whether to load the trained model and the data that has been self-labeled '
283-
'from a previous run, if available. This is useful if a process can get '
284-
'preempted or interrupted.')
281+
'load_from_checkpoint', False,
282+
'Whether to load the trained model and the data that has been self-labeled '
283+
'from a previous run, if available. This is useful if a process can get '
284+
'preempted or interrupted.')
285285

286286

287287
def parse_layers_string(layers_string):

0 commit comments

Comments
 (0)