Add conditioning-tutorial checkpoint: joint_model

.gitattributes

@@ -33,3 +33,4 @@ 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
+conditioning_and_guidance/joint_model/1500/opt_state/ocdbt.process_0/d/04b2b6c8f458848a158e39ba5b736356 filter=lfs diff=lfs merge=lfs -text

conditioning_and_guidance/joint_model/1500/_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": 1785242200039314000, "commit_timestamp_nsecs": 1785242200114307000, "custom_metadata": {"num_steps": 1500}}

conditioning_and_guidance/joint_model/1500/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/1500/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/1500/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/1500/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',)": {"key_metadata": [{"key": "1", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}

conditioning_and_guidance/joint_model/1500/opt_state/ocdbt.process_0/d/04b2b6c8f458848a158e39ba5b736356

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

conditioning_and_guidance/joint_model/1500/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}