File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed
neural_structured_learning/research/gam/experiments Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change 194194flags .DEFINE_integer (
195195 'batch_size_cls' , 512 , 'Batch size for classification model.' )
196196flags .DEFINE_float (
197- 'gradient_clip' , None ,
197+ 'gradient_clip' , 10 ,
198198 'The gradient clipping global norm value. If None, no clipping is done.' )
199199flags .DEFINE_integer (
200200 'logging_step_cls' , 200 ,
278278 'optimizer' , 'adam' ,
279279 'Which optimizer to use. Valid options are `adam`, `amsgrad`.' )
280280flags .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
287287def parse_layers_string (layers_string ):
You can’t perform that action at this time.
0 commit comments