BackupAndRestore is not experimental any more.

Johannes Ballé · copybara-github · commit cbe6ce91db68 · 2022-02-16T14:28:36.000-08:00
PiperOrigin-RevId: 429140913
Change-Id: Ibb5f97f80cf74412949d916d4289c6330790d962
diff --git a/models/bls2017.py b/models/bls2017.py
@@ -263,7 +263,7 @@ def train(args):
           tf.keras.callbacks.TensorBoard(
               log_dir=args.train_path,
               histogram_freq=1, update_freq="epoch"),
-          tf.keras.callbacks.experimental.BackupAndRestore(args.train_path),
+          tf.keras.callbacks.BackupAndRestore(args.train_path),
       ],
       verbose=int(args.verbose),
   )
diff --git a/models/bmshj2018.py b/models/bmshj2018.py
@@ -339,7 +339,7 @@ def train(args):
           tf.keras.callbacks.TensorBoard(
               log_dir=args.train_path,
               histogram_freq=1, update_freq="epoch"),
-          tf.keras.callbacks.experimental.BackupAndRestore(args.train_path),
+          tf.keras.callbacks.BackupAndRestore(args.train_path),
       ],
       verbose=int(args.verbose),
   )
diff --git a/models/ms2020.py b/models/ms2020.py
@@ -510,7 +510,7 @@ def train(args):
           tf.keras.callbacks.TensorBoard(
               log_dir=args.train_path,
               histogram_freq=1, update_freq="epoch"),
-          tf.keras.callbacks.experimental.BackupAndRestore(args.train_path),
+          tf.keras.callbacks.BackupAndRestore(args.train_path),
       ],
       verbose=int(args.verbose),
   )
diff --git a/models/toy_sources/toy_sources.ipynb b/models/toy_sources/toy_sources.ipynb
@@ -613,7 +613,7 @@
         "    tf.keras.callbacks.ModelCheckpoint(\n",
         "        work_path + \"/checkpoints/ckpt-{epoch:04d}\",\n",
         "        save_weights_only=True),\n",
-        "    tf.keras.callbacks.experimental.BackupAndRestore(\n",
+        "    tf.keras.callbacks.BackupAndRestore(\n",
         "        work_path + \"/backup\"),\n",
         "    tf.keras.callbacks.LearningRateScheduler(lr_scheduler),\n",
         "    AlphaScheduler(alpha_scheduler),\n",
@@ -687,7 +687,7 @@
         "    tf.keras.callbacks.ModelCheckpoint(\n",
         "        work_path + \"/checkpoints/ckpt-{epoch:04d}\",\n",
         "        save_weights_only=True),\n",
-        "    tf.keras.callbacks.experimental.BackupAndRestore(\n",
+        "    tf.keras.callbacks.BackupAndRestore(\n",
         "        work_path + \"/backup\"),\n",
         "    tf.keras.callbacks.LearningRateScheduler(lr_scheduler),\n",
         "    AlphaScheduler(alpha_scheduler),\n",
@@ -776,7 +776,7 @@
         "    tf.keras.callbacks.ModelCheckpoint(\n",
         "        work_path + \"/checkpoints/ckpt-{epoch:04d}\",\n",
         "        save_weights_only=True),\n",
-        "    tf.keras.callbacks.experimental.BackupAndRestore(\n",
+        "    tf.keras.callbacks.BackupAndRestore(\n",
         "        work_path + \"/backup\"),\n",
         "    tf.keras.callbacks.LearningRateScheduler(lr_scheduler),\n",
         "    AlphaScheduler(alpha_scheduler),\n",
@@ -850,7 +850,7 @@
         "    tf.keras.callbacks.ModelCheckpoint(\n",
         "        work_path + \"/checkpoints/ckpt-{epoch:04d}\",\n",
         "        save_weights_only=True),\n",
-        "    tf.keras.callbacks.experimental.BackupAndRestore(\n",
+        "    tf.keras.callbacks.BackupAndRestore(\n",
         "        work_path + \"/backup\"),\n",
         "    tf.keras.callbacks.LearningRateScheduler(lr_scheduler),\n",
         "    AlphaScheduler(alpha_scheduler),\n",
@@ -928,7 +928,7 @@
         "    tf.keras.callbacks.ModelCheckpoint(\n",
         "        work_path + \"/checkpoints/ckpt-{epoch:04d}\",\n",
         "        save_weights_only=True),\n",
-        "    tf.keras.callbacks.experimental.BackupAndRestore(\n",
+        "    tf.keras.callbacks.BackupAndRestore(\n",
         "        work_path + \"/backup\"),\n",
         "    tf.keras.callbacks.LearningRateScheduler(lr_scheduler),\n",
         "    AlphaScheduler(alpha_scheduler),\n",
@@ -1002,7 +1002,7 @@
         "    tf.keras.callbacks.ModelCheckpoint(\n",
         "        work_path + \"/checkpoints/ckpt-{epoch:04d}\",\n",
         "        save_weights_only=True),\n",
-        "    tf.keras.callbacks.experimental.BackupAndRestore(\n",
+        "    tf.keras.callbacks.BackupAndRestore(\n",
         "        work_path + \"/backup\"),\n",
         "    tf.keras.callbacks.LearningRateScheduler(lr_scheduler),\n",
         "    AlphaScheduler(alpha_scheduler),\n",
@@ -1072,7 +1072,7 @@
         "    tf.keras.callbacks.ModelCheckpoint(\n",
         "        work_path + \"/checkpoints/ckpt-{epoch:04d}\",\n",
         "        save_weights_only=True),\n",
-        "    tf.keras.callbacks.experimental.BackupAndRestore(\n",
+        "    tf.keras.callbacks.BackupAndRestore(\n",
         "        work_path + \"/backup\"),\n",
         "    tf.keras.callbacks.LearningRateScheduler(lr_scheduler),\n",
         "    AlphaScheduler(alpha_scheduler),\n",
@@ -1140,7 +1140,7 @@
         "    tf.keras.callbacks.ModelCheckpoint(\n",
         "        work_path + \"/checkpoints/ckpt-{epoch:04d}\",\n",
         "        save_weights_only=True),\n",
-        "    tf.keras.callbacks.experimental.BackupAndRestore(\n",
+        "    tf.keras.callbacks.BackupAndRestore(\n",
         "        work_path + \"/backup\"),\n",
         "    tf.keras.callbacks.LearningRateScheduler(lr_scheduler),\n",
         "    AlphaScheduler(alpha_scheduler),\n",

Original file line number	Diff line number	Diff line change
`@@ -263,7 +263,7 @@ def train(args):`
`263`	`263`	`tf.keras.callbacks.TensorBoard(`
`264`	`264`	`log_dir=args.train_path,`
`265`	`265`	`histogram_freq=1, update_freq="epoch"),`
`266`		`- tf.keras.callbacks.experimental.BackupAndRestore(args.train_path),`
	`266`	`+ tf.keras.callbacks.BackupAndRestore(args.train_path),`
`267`	`267`	`],`
`268`	`268`	`verbose=int(args.verbose),`
`269`	`269`	`)`
Original file line number	Diff line number	Diff line change
`@@ -339,7 +339,7 @@ def train(args):`
`339`	`339`	`tf.keras.callbacks.TensorBoard(`
`340`	`340`	`log_dir=args.train_path,`
`341`	`341`	`histogram_freq=1, update_freq="epoch"),`
`342`		`- tf.keras.callbacks.experimental.BackupAndRestore(args.train_path),`
	`342`	`+ tf.keras.callbacks.BackupAndRestore(args.train_path),`
`343`	`343`	`],`
`344`	`344`	`verbose=int(args.verbose),`
`345`	`345`	`)`
Original file line number	Diff line number	Diff line change
`@@ -510,7 +510,7 @@ def train(args):`
`510`	`510`	`tf.keras.callbacks.TensorBoard(`
`511`	`511`	`log_dir=args.train_path,`
`512`	`512`	`histogram_freq=1, update_freq="epoch"),`
`513`		`- tf.keras.callbacks.experimental.BackupAndRestore(args.train_path),`
	`513`	`+ tf.keras.callbacks.BackupAndRestore(args.train_path),`
`514`	`514`	`],`
`515`	`515`	`verbose=int(args.verbose),`
`516`	`516`	`)`