From 18a8c20362b71c3a90361240832380348fcb7cfc Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Tue, 19 Sep 2023 21:53:16 +0000 Subject: [PATCH] pylint --- submission_runner.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/submission_runner.py b/submission_runner.py index fb8df198c..af3741812 100644 --- a/submission_runner.py +++ b/submission_runner.py @@ -134,7 +134,7 @@ True, 'Whether or not to checkpoint the model at every eval.') flags.DEFINE_integer('rng_seed', - None, + None, 'Value of rng seed. If None, a random seed will' 'be generated from hardware.') FLAGS = flags.FLAGS @@ -177,7 +177,7 @@ def train_once( update_params: spec.UpdateParamsFn, data_selection: spec.DataSelectionFn, hyperparameters: Optional[spec.Hyperparameters], - rng_seed: int, + rng_seed: int, rng: spec.RandomState, profiler: Profiler, max_global_steps: int = None, @@ -534,7 +534,7 @@ def score_submission_on_workload(workload: spec.Workload, data_dir, imagenet_v2_data_dir, init_optimizer_state, update_params, data_selection, - hyperparameters, + hyperparameters, rng_seed, rng, profiler,