Upload 13 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+state/ocdbt.process_0/d/22dfc03d0f10589a3e6a59ac00f4f85a filter=lfs diff=lfs merge=lfs -text
+state/ocdbt.process_0/d/df6647504b1c6c04dc6d43d3f4985f71 filter=lfs diff=lfs merge=lfs -text

config.json

@@ -0,0 +1,40 @@
+{
+  "architecture": "resnet18",
+  "num_classes": 1000,
+  "num_features": 512,
+  "global_pool": "avg",
+  "pretrained_cfg": {
+    "architecture": "resnet18",
+    "input_size": [
+      224,
+      224,
+      3
+    ],
+    "interpolation": "bicubic",
+    "crop_pct": 0.95,
+    "crop_mode": "center",
+    "mean": [
+      0.485,
+      0.456,
+      0.406
+    ],
+    "std": [
+      0.229,
+      0.224,
+      0.225
+    ],
+    "num_classes": 1000,
+    "pool_size": [
+      7,
+      7
+    ],
+    "first_conv": "conv1",
+    "classifier": "fc",
+    "test_input_size": [
+      288,
+      288,
+      3
+    ],
+    "test_crop_pct": 1.0
+  }
+}

state/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1775590172333354901, "commit_timestamp_nsecs": 1775590172959265887, "custom_metadata": {}}

state/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('bn1', 'bias')": {"key_metadata": [{"key": "bn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('bn1', 'mean')": {"key_metadata": [{"key": "bn1", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('bn1', 'scale')": {"key_metadata": [{"key": "bn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('bn1', 'var')": {"key_metadata": [{"key": "bn1", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('conv1', 'kernel')": {"key_metadata": [{"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [7, 7, 3, 64]}}, "('fc', 'fc', 'bias')": {"key_metadata": [{"key": "fc", "key_type": 2}, {"key": "fc", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1000]}}, "('fc', 'fc', 'kernel')": {"key_metadata": [{"key": "fc", "key_type": 2}, {"key": "fc", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512, 1000]}}, "('stage_modules', '0', 'layers', '0', 'bn1', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '0', 'bn1', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '0', 'bn1', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '0', 'bn1', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '0', 'bn2', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '0', 'bn2', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '0', 'bn2', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '0', 'bn2', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '0', 'conv1', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 64, 64]}}, "('stage_modules', '0', 'layers', '0', 'conv2', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 64, 64]}}, "('stage_modules', '0', 'layers', '1', 'bn1', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '1', 'bn1', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '1', 'bn1', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '1', 'bn1', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '1', 'bn2', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '1', 'bn2', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '1', 'bn2', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '1', 'bn2', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [64]}}, "('stage_modules', '0', 'layers', '1', 'conv1', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 64, 64]}}, "('stage_modules', '0', 'layers', '1', 'conv2', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 64, 64]}}, "('stage_modules', '1', 'layers', '0', 'bn1', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'bn1', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'bn1', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'bn1', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'bn2', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'bn2', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'bn2', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'bn2', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'conv1', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 64, 128]}}, "('stage_modules', '1', 'layers', '0', 'conv2', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 128, 128]}}, "('stage_modules', '1', 'layers', '0', 'downsample', 'bn', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'downsample', 'bn', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'downsample', 'bn', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'downsample', 'bn', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '0', 'downsample', 'conv', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 1, 64, 128]}}, "('stage_modules', '1', 'layers', '1', 'bn1', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '1', 'bn1', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '1', 'bn1', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '1', 'bn1', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '1', 'bn2', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '1', 'bn2', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '1', 'bn2', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '1', 'bn2', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [128]}}, "('stage_modules', '1', 'layers', '1', 'conv1', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 128, 128]}}, "('stage_modules', '1', 'layers', '1', 'conv2', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 128, 128]}}, "('stage_modules', '2', 'layers', '0', 'bn1', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'bn1', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'bn1', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'bn1', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'bn2', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'bn2', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'bn2', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'bn2', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'conv1', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 128, 256]}}, "('stage_modules', '2', 'layers', '0', 'conv2', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 256, 256]}}, "('stage_modules', '2', 'layers', '0', 'downsample', 'bn', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'downsample', 'bn', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'downsample', 'bn', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'downsample', 'bn', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '0', 'downsample', 'conv', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 1, 128, 256]}}, "('stage_modules', '2', 'layers', '1', 'bn1', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '1', 'bn1', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '1', 'bn1', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '1', 'bn1', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '1', 'bn2', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '1', 'bn2', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '1', 'bn2', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '1', 'bn2', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [256]}}, "('stage_modules', '2', 'layers', '1', 'conv1', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 256, 256]}}, "('stage_modules', '2', 'layers', '1', 'conv2', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 256, 256]}}, "('stage_modules', '3', 'layers', '0', 'bn1', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'bn1', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'bn1', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'bn1', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'bn2', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'bn2', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'bn2', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'bn2', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'conv1', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 256, 512]}}, "('stage_modules', '3', 'layers', '0', 'conv2', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 512, 512]}}, "('stage_modules', '3', 'layers', '0', 'downsample', 'bn', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'downsample', 'bn', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'downsample', 'bn', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'downsample', 'bn', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "bn", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '0', 'downsample', 'conv', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "downsample", "key_type": 2}, {"key": "conv", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [1, 1, 256, 512]}}, "('stage_modules', '3', 'layers', '1', 'bn1', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '1', 'bn1', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '1', 'bn1', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '1', 'bn1', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn1", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '1', 'bn2', 'bias')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '1', 'bn2', 'mean')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "mean", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '1', 'bn2', 'scale')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "scale", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '1', 'bn2', 'var')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "bn2", "key_type": 2}, {"key": "var", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [512]}}, "('stage_modules', '3', 'layers', '1', 'conv1', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "conv1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 512, 512]}}, "('stage_modules', '3', 'layers', '1', 'conv2', 'kernel')": {"key_metadata": [{"key": "stage_modules", "key_type": 2}, {"key": "3", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "1", "key_type": 2}, {"key": "conv2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [3, 3, 512, 512]}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

state/_sharding

@@ -0,0 +1 @@
+{"Y29udjEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","Ym4xLm1lYW4=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","Ym4xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","Ym4xLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","Ym4xLnZhcg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","ZmMuZmMuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","ZmMuZmMua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmJuMS52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmJuMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmJuMS5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmJuMS5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmJuMi52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmJuMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmJuMi5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmJuMi5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmNvbnYxLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4wLmNvbnYyLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmJuMS52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmJuMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmJuMS5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmJuMS5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmJuMi52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmJuMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmJuMi5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmJuMi5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmNvbnYxLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4wLmxheWVycy4xLmNvbnYyLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmJuMS52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmJuMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmJuMS5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmJuMS5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmJuMi52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmJuMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmJuMi5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmJuMi5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmNvbnYxLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmNvbnYyLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmRvd25zYW1wbGUuY29udi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmRvd25zYW1wbGUuYm4uYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmRvd25zYW1wbGUuYm4ubWVhbg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmRvd25zYW1wbGUuYm4uc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4wLmRvd25zYW1wbGUuYm4udmFy":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmJuMS52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmJuMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmJuMS5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmJuMS5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmJuMi52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmJuMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmJuMi5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmJuMi5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmNvbnYxLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4xLmxheWVycy4xLmNvbnYyLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmJuMS52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmJuMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmJuMS5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmJuMS5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmJuMi52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmJuMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmJuMi5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmJuMi5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmNvbnYxLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmNvbnYyLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmRvd25zYW1wbGUuY29udi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmRvd25zYW1wbGUuYm4uYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmRvd25zYW1wbGUuYm4ubWVhbg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmRvd25zYW1wbGUuYm4uc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4wLmRvd25zYW1wbGUuYm4udmFy":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmJuMS52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmJuMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmJuMS5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmJuMS5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmJuMi52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmJuMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmJuMi5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmJuMi5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmNvbnYxLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4yLmxheWVycy4xLmNvbnYyLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmJuMS52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmJuMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmJuMS5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmJuMS5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmJuMi52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmJuMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmJuMi5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmJuMi5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmNvbnYxLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmNvbnYyLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmRvd25zYW1wbGUuY29udi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmRvd25zYW1wbGUuYm4uYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmRvd25zYW1wbGUuYm4ubWVhbg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmRvd25zYW1wbGUuYm4uc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4wLmRvd25zYW1wbGUuYm4udmFy":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmJuMS52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmJuMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmJuMS5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmJuMS5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmJuMi52YXI=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmJuMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmJuMi5tZWFu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmJuMi5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmNvbnYxLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}","c3RhZ2VfbW9kdWxlcy4zLmxheWVycy4xLmNvbnYyLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TFRT_CPU_0\"}"}

state/array_metadatas/process_0

@@ -0,0 +1 @@
+{"array_metadatas": [{"array_metadata": {"param_name": "bn1.bias", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "bn1.mean", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "bn1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "bn1.var", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "conv1.kernel", "write_shape": [7, 7, 3, 64], "chunk_shape": [7, 7, 3, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "fc.fc.bias", "write_shape": [1000], "chunk_shape": [1000], "ext_metadata": null}}, {"array_metadata": {"param_name": "fc.fc.kernel", "write_shape": [512, 1000], "chunk_shape": [512, 1000], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.bn1.bias", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.bn1.mean", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.bn1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.bn1.var", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.bn2.bias", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.bn2.mean", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.bn2.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.bn2.var", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.conv1.kernel", "write_shape": [3, 3, 64, 64], "chunk_shape": [3, 3, 64, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.0.conv2.kernel", "write_shape": [3, 3, 64, 64], "chunk_shape": [3, 3, 64, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.bn1.bias", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.bn1.mean", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.bn1.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.bn1.var", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.bn2.bias", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.bn2.mean", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.bn2.scale", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.bn2.var", "write_shape": [64], "chunk_shape": [64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.conv1.kernel", "write_shape": [3, 3, 64, 64], "chunk_shape": [3, 3, 64, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.0.layers.1.conv2.kernel", "write_shape": [3, 3, 64, 64], "chunk_shape": [3, 3, 64, 64], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.bn1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.bn1.mean", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.bn1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.bn1.var", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.bn2.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.bn2.mean", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.bn2.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.bn2.var", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.conv1.kernel", "write_shape": [3, 3, 64, 128], "chunk_shape": [3, 3, 64, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.conv2.kernel", "write_shape": [3, 3, 128, 128], "chunk_shape": [3, 3, 128, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.downsample.bn.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.downsample.bn.mean", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.downsample.bn.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.downsample.bn.var", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.0.downsample.conv.kernel", "write_shape": [1, 1, 64, 128], "chunk_shape": [1, 1, 64, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.bn1.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.bn1.mean", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.bn1.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.bn1.var", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.bn2.bias", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.bn2.mean", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.bn2.scale", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.bn2.var", "write_shape": [128], "chunk_shape": [128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.conv1.kernel", "write_shape": [3, 3, 128, 128], "chunk_shape": [3, 3, 128, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.1.layers.1.conv2.kernel", "write_shape": [3, 3, 128, 128], "chunk_shape": [3, 3, 128, 128], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.bn1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.bn1.mean", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.bn1.scale", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.bn1.var", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.bn2.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.bn2.mean", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.bn2.scale", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.bn2.var", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.conv1.kernel", "write_shape": [3, 3, 128, 256], "chunk_shape": [3, 3, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.conv2.kernel", "write_shape": [3, 3, 256, 256], "chunk_shape": [3, 3, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.downsample.bn.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.downsample.bn.mean", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.downsample.bn.scale", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.downsample.bn.var", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.0.downsample.conv.kernel", "write_shape": [1, 1, 128, 256], "chunk_shape": [1, 1, 128, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.bn1.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.bn1.mean", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.bn1.scale", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.bn1.var", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.bn2.bias", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.bn2.mean", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.bn2.scale", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.bn2.var", "write_shape": [256], "chunk_shape": [256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.conv1.kernel", "write_shape": [3, 3, 256, 256], "chunk_shape": [3, 3, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.2.layers.1.conv2.kernel", "write_shape": [3, 3, 256, 256], "chunk_shape": [3, 3, 256, 256], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.bn1.bias", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.bn1.mean", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.bn1.scale", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.bn1.var", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.bn2.bias", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.bn2.mean", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.bn2.scale", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.bn2.var", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.conv1.kernel", "write_shape": [3, 3, 256, 512], "chunk_shape": [3, 3, 256, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.conv2.kernel", "write_shape": [3, 3, 512, 512], "chunk_shape": [3, 3, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.downsample.bn.bias", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.downsample.bn.mean", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.downsample.bn.scale", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.downsample.bn.var", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.0.downsample.conv.kernel", "write_shape": [1, 1, 256, 512], "chunk_shape": [1, 1, 256, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.bn1.bias", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.bn1.mean", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.bn1.scale", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.bn1.var", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.bn2.bias", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.bn2.mean", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.bn2.scale", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.bn2.var", "write_shape": [512], "chunk_shape": [512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.conv1.kernel", "write_shape": [3, 3, 512, 512], "chunk_shape": [3, 3, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "stage_modules.3.layers.1.conv2.kernel", "write_shape": [3, 3, 512, 512], "chunk_shape": [3, 3, 512, 512], "ext_metadata": null}}]}

state/ocdbt.process_0/d/22dfc03d0f10589a3e6a59ac00f4f85a

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:421f40e5ef1edf508e8c8b996404fe84724ccc42ed4715b308a1f4d964251978
+size 41422848

state/ocdbt.process_0/d/df6647504b1c6c04dc6d43d3f4985f71

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bc82370a4a82029a73f30255875ff7dfb2291e076010ddbe34bbb46dce1c5e24
+size 2035086