lprimeau commited on
Commit
62f9978
·
verified ·
1 Parent(s): 504e902

Upload folder using huggingface_hub

Browse files
_CHECKPOINT_METADATA ADDED
@@ -0,0 +1 @@
 
 
1
+ {"item_handlers": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler", "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1761285556965267633, "commit_timestamp_nsecs": 1761285557070173534, "custom_metadata": {}}
_METADATA ADDED
@@ -0,0 +1 @@
 
 
1
+ {"tree_metadata": {"('conv1', 'bias')": {"key_metadata": [{"key": "conv1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [4]}}, "('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": [2, 2, 1, 4]}}, "('linear1', 'bias')": {"key_metadata": [{"key": "linear1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [10]}}, "('linear1', 'kernel')": {"key_metadata": [{"key": "linear1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false, "write_shape": [144, 10]}}}, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}
_sharding ADDED
@@ -0,0 +1 @@
 
 
1
+ {"Y29udjEuYmlhcw==":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1], \"axis_names\": [\"S\"], \"axis_types\": [\"AxisType.Auto\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [{\"id\": 0}]}}","Y29udjEua2VybmVs":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1], \"axis_names\": [\"S\"], \"axis_types\": [\"AxisType.Auto\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [{\"id\": 0}]}}","bGluZWFyMS5iaWFz":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1], \"axis_names\": [\"S\"], \"axis_types\": [\"AxisType.Auto\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [{\"id\": 0}]}}","bGluZWFyMS5rZXJuZWw=":"{\"sharding_type\": \"NamedSharding\", \"shape\": [1], \"axis_names\": [\"S\"], \"axis_types\": [\"AxisType.Auto\"], \"partition_spec\": [], \"device_mesh\": {\"mesh\": [{\"id\": 0}]}}"}
array_metadatas/process_0 ADDED
@@ -0,0 +1 @@
 
 
1
+ {"array_metadatas": [{"array_metadata": {"param_name": "conv1.bias", "write_shape": [4], "chunk_shape": [4], "ext_metadata": null}}, {"array_metadata": {"param_name": "conv1.kernel", "write_shape": [2, 2, 1, 4], "chunk_shape": [2, 2, 1, 4], "ext_metadata": null}}, {"array_metadata": {"param_name": "linear1.bias", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "linear1.kernel", "write_shape": [144, 10], "chunk_shape": [144, 10], "ext_metadata": null}}]}
d/0a94fc886df7cf21767eb4d210db87d6 ADDED
Binary file (614 Bytes). View file
 
manifest.ocdbt ADDED
Binary file (114 Bytes). View file
 
ocdbt.process_0/d/0474af1c3ec249c72876071e48c16120 ADDED
Binary file (10.9 kB). View file
 
ocdbt.process_0/d/3f1fa50644c3f3a753f56fb0c1305f3a ADDED
Binary file (598 Bytes). View file
 
ocdbt.process_0/d/9c9c16a1515a3ab9a400d8c2c05a7dc4 ADDED
Binary file (249 Bytes). View file
 
ocdbt.process_0/d/ce6f52b188aaa1d2f283c079387fb439 ADDED
Binary file (185 Bytes). View file
 
ocdbt.process_0/manifest.ocdbt ADDED
Binary file (244 Bytes). View file