Add conditioning-tutorial checkpoint: joint_model

.gitattributes

@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 conditioning_and_guidance/joint_model/1500/opt_state/ocdbt.process_0/d/04b2b6c8f458848a158e39ba5b736356 filter=lfs diff=lfs merge=lfs -text
 conditioning_and_guidance/context_model/1500/opt_state/ocdbt.process_0/d/07544e5937eab293eb793c6581913fc0 filter=lfs diff=lfs merge=lfs -text
+conditioning_and_guidance/joint_model/3000/opt_state/ocdbt.process_0/d/2be637e12a96f25ef46554e11f991918 filter=lfs diff=lfs merge=lfs -text

conditioning_and_guidance/joint_model/3000/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"item_handlers": {"dataset_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "ema_params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "extra": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "opt_state": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler", "params": "orbax.checkpoint._src.handlers.pytree_checkpoint_handler.PyTreeCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1785786862497780000, "commit_timestamp_nsecs": 1785786862567612000, "custom_metadata": {"num_steps": 3000}}

conditioning_and_guidance/joint_model/3000/dataset_state/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('next_index',)": {"key_metadata": [{"key": "next_index", "key_type": 2}], "value_metadata": {"value_type": "scalar", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

conditioning_and_guidance/joint_model/3000/ema_params/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('network', 'heads', 'coordinates', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'coordinates', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'index', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "index", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'index', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "index", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

conditioning_and_guidance/joint_model/3000/extra/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('ema_decay',)": {"key_metadata": [{"key": "ema_decay", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('num_steps',)": {"key_metadata": [{"key": "num_steps", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('rng_key',)": {"key_metadata": [{"key": "rng_key", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

conditioning_and_guidance/joint_model/3000/opt_state/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('0', 'count')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'heads', 'coordinates', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'heads', 'coordinates', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'heads', 'index', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "index", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'heads', 'index', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "index", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'trunk', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'trunk', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'trunk', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'mu', 'network', 'trunk', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'heads', 'coordinates', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'heads', 'coordinates', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'heads', 'index', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "index", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'heads', 'index', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "index", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'trunk', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'trunk', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'trunk', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('0', 'nu', 'network', 'trunk', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('1', 'count')": {"key_metadata": [{"key": "1", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

conditioning_and_guidance/joint_model/3000/opt_state/ocdbt.process_0/d/2be637e12a96f25ef46554e11f991918

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

conditioning_and_guidance/joint_model/3000/params/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('network', 'heads', 'coordinates', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'coordinates', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'index', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "index", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'index', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "index", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}