Upload folder using huggingface_hub

.gitattributes

@@ -37,3 +37,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 checkpoints/39911424/default/ocdbt.process_0/d/7d33210c5d5cb58d232a0389728c1570 filter=lfs diff=lfs merge=lfs -text
 checkpoints/54878208/default/ocdbt.process_0/d/304e635d36a8e499ee67dcba9c427894 filter=lfs diff=lfs merge=lfs -text
 checkpoints/74833920/default/ocdbt.process_0/d/a130feea07cfc515fedce1caf8a8ad46 filter=lfs diff=lfs merge=lfs -text
+checkpoints/94789632/default/ocdbt.process_0/d/764408dd0353b32b2d8425bfb4cf4b99 filter=lfs diff=lfs merge=lfs -text

checkpoints/94789632/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"init_timestamp_nsecs": 1729023981311029191, "commit_timestamp_nsecs": 1729023981434618040}

checkpoints/94789632/default/_METADATA

@@ -0,0 +1 @@
+{"tree_metadata": {"('learner_state', 'params', 'ac_params', 'params', 'action_head', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "action_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'ac_params', 'params', 'action_head', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "action_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'ac_params', 'params', 'critic_head', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "critic_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'ac_params', 'params', 'critic_head', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "critic_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'ac_params', 'params', 'shared_torso', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'ac_params', 'params', 'shared_torso', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'ac_params', 'params', 'shared_torso', 'Dense_1', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'ac_params', 'params', 'shared_torso', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'ac_params', 'params', 'shared_torso', 'Dense_2', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'ac_params', 'params', 'shared_torso', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'action_head', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "action_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'action_head', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "action_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'critic_head', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "critic_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'critic_head', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "critic_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'shared_torso', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'shared_torso', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'shared_torso', 'Dense_1', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'shared_torso', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'shared_torso', 'Dense_2', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'target_ac_params', 'params', 'shared_torso', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "target_ac_params", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'dual_params', 'log_temperature')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "dual_params", "key_type": 2}, {"key": "log_temperature", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'params', 'dual_params', 'log_alpha')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "dual_params", "key_type": 2}, {"key": "log_alpha", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '0')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "0", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'count')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'action_head', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "action_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'action_head', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "action_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'critic_head', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "critic_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'critic_head', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "critic_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'shared_torso', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'shared_torso', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'shared_torso', 'Dense_1', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'shared_torso', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'shared_torso', 'Dense_2', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'mu', 'params', 'shared_torso', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'action_head', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "action_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'action_head', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "action_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'critic_head', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "critic_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'critic_head', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "critic_head", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'shared_torso', 'Dense_0', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'shared_torso', 'Dense_0', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_0", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'shared_torso', 'Dense_1', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'shared_torso', 'Dense_1', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_1", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'shared_torso', 'Dense_2', 'bias')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "bias", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '0', 'nu', 'params', 'shared_torso', 'Dense_2', 'kernel')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "params", "key_type": 2}, {"key": "shared_torso", "key_type": 2}, {"key": "Dense_2", "key_type": 2}, {"key": "kernel", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '1')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "1", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('learner_state', 'opt_states', 'ac_opt_state', '1', '2')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "ac_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "2", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('learner_state', 'opt_states', 'dual_opt_state', '0')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "dual_opt_state", "key_type": 2}, {"key": "0", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('learner_state', 'opt_states', 'dual_opt_state', '1', '0', 'count')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "dual_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'dual_opt_state', '1', '0', 'mu', 'log_temperature')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "dual_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "log_temperature", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'dual_opt_state', '1', '0', 'mu', 'log_alpha')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "dual_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "mu", "key_type": 2}, {"key": "log_alpha", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'dual_opt_state', '1', '0', 'nu', 'log_temperature')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "dual_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "log_temperature", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'dual_opt_state', '1', '0', 'nu', 'log_alpha')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "dual_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "0", "key_type": 1}, {"key": "nu", "key_type": 2}, {"key": "log_alpha", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'opt_states', 'dual_opt_state', '1', '1')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "opt_states", "key_type": 2}, {"key": "dual_opt_state", "key_type": 2}, {"key": "1", "key_type": 1}, {"key": "1", "key_type": 1}], "value_metadata": {"value_type": "None", "skip_deserialize": true}}, "('learner_state', 'buffer_state', 'current_index')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "current_index", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'done')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "done", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'truncated')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "truncated", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'action')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "action", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'sampled_actions')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "sampled_actions", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'reward')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "reward", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'search_value')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "search_value", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'search_policy')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "search_policy", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'obs', 'agent_view')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "obs", "key_type": 2}, {"key": "agent_view", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'obs', 'action_mask')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "obs", "key_type": 2}, {"key": "action_mask", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'obs', 'step_count')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "obs", "key_type": 2}, {"key": "step_count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'info', 'episode_length')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "info", "key_type": 2}, {"key": "episode_length", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'info', 'episode_return')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "info", "key_type": 2}, {"key": "episode_return", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'info', 'is_terminal_step')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "info", "key_type": 2}, {"key": "is_terminal_step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'experience', 'advantages')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "experience", "key_type": 2}, {"key": "advantages", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'buffer_state', 'is_full')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "buffer_state", "key_type": 2}, {"key": "is_full", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'key')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "key", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'env_state', 'gymnax_env_state', 'time')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "gymnax_env_state", "key_type": 2}, {"key": "time", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'env_state', 'gymnax_env_state', 'x')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "gymnax_env_state", "key_type": 2}, {"key": "x", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'env_state', 'gymnax_env_state', 'x_dot')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "gymnax_env_state", "key_type": 2}, {"key": "x_dot", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'env_state', 'gymnax_env_state', 'theta')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "gymnax_env_state", "key_type": 2}, {"key": "theta", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'env_state', 'gymnax_env_state', 'theta_dot')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "gymnax_env_state", "key_type": 2}, {"key": "theta_dot", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'env_state', 'key')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "key", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'env_state', 'step_count')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "step_count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'key')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "key", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'running_count_episode_return')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "running_count_episode_return", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'running_count_episode_length')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "running_count_episode_length", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'episode_return')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "episode_return", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'env_state', 'episode_length')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "env_state", "key_type": 2}, {"key": "episode_length", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'discount')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "discount", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'extras', 'episode_metrics', 'episode_length')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "extras", "key_type": 2}, {"key": "episode_metrics", "key_type": 2}, {"key": "episode_length", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'extras', 'episode_metrics', 'episode_return')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "extras", "key_type": 2}, {"key": "episode_metrics", "key_type": 2}, {"key": "episode_return", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'extras', 'episode_metrics', 'is_terminal_step')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "extras", "key_type": 2}, {"key": "episode_metrics", "key_type": 2}, {"key": "is_terminal_step", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'extras', 'next_obs', 'agent_view')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "extras", "key_type": 2}, {"key": "next_obs", "key_type": 2}, {"key": "agent_view", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'extras', 'next_obs', 'action_mask')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "extras", "key_type": 2}, {"key": "next_obs", "key_type": 2}, {"key": "action_mask", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'extras', 'next_obs', 'step_count')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "extras", "key_type": 2}, {"key": "next_obs", "key_type": 2}, {"key": "step_count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'observation', 'agent_view')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "observation", "key_type": 2}, {"key": "agent_view", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'observation', 'action_mask')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "observation", "key_type": 2}, {"key": "action_mask", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'observation', 'step_count')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "observation", "key_type": 2}, {"key": "step_count", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'reward')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "reward", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}, "('learner_state', 'timestep', 'step_type')": {"key_metadata": [{"key": "learner_state", "key_type": 2}, {"key": "timestep", "key_type": 2}, {"key": "step_type", "key_type": 2}], "value_metadata": {"value_type": "jax.Array", "skip_deserialize": false}}}, "use_zarr3": false}

checkpoints/94789632/default/_sharding

@@ -0,0 +1 @@
+{"bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5kaXNjb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5leHRyYXMuZXBpc29kZV9tZXRyaWNzLmVwaXNvZGVfbGVuZ3Ro":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5leHRyYXMuZXBpc29kZV9tZXRyaWNzLmVwaXNvZGVfcmV0dXJu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5leHRyYXMuZXBpc29kZV9tZXRyaWNzLmlzX3Rlcm1pbmFsX3N0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5leHRyYXMubmV4dF9vYnMuYWN0aW9uX21hc2s=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5leHRyYXMubmV4dF9vYnMuYWdlbnRfdmlldw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5leHRyYXMubmV4dF9vYnMuc3RlcF9jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5vYnNlcnZhdGlvbi5hY3Rpb25fbWFzaw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5vYnNlcnZhdGlvbi5hZ2VudF92aWV3":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5vYnNlcnZhdGlvbi5zdGVwX2NvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5yZXdhcmQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS50aW1lc3RlcC5zdGVwX3R5cGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuY3VycmVudF9pbmRleA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS50cnVuY2F0ZWQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5hY3Rpb24=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5hZHZhbnRhZ2Vz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5kb25l":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5pbmZvLmVwaXNvZGVfbGVuZ3Ro":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5pbmZvLmVwaXNvZGVfcmV0dXJu":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5pbmZvLmlzX3Rlcm1pbmFsX3N0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5vYnMuYWN0aW9uX21hc2s=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5vYnMuYWdlbnRfdmlldw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5vYnMuc3RlcF9jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5yZXdhcmQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5zYW1wbGVkX2FjdGlvbnM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5zZWFyY2hfcG9saWN5":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuZXhwZXJpZW5jZS5zZWFyY2hfdmFsdWU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5idWZmZXJfc3RhdGUuaXNfZnVsbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUuZW52X3N0YXRlLmd5bW5heF9lbnZfc3RhdGUudGhldGE=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUuZW52X3N0YXRlLmd5bW5heF9lbnZfc3RhdGUudGhldGFfZG90":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUuZW52X3N0YXRlLmd5bW5heF9lbnZfc3RhdGUudGltZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUuZW52X3N0YXRlLmd5bW5heF9lbnZfc3RhdGUueA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUuZW52X3N0YXRlLmd5bW5heF9lbnZfc3RhdGUueF9kb3Q=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUuZW52X3N0YXRlLmtleQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUuZW52X3N0YXRlLnN0ZXBfY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUuZXBpc29kZV9sZW5ndGg=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUuZXBpc29kZV9yZXR1cm4=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUua2V5":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUucnVubmluZ19jb3VudF9lcGlzb2RlX2xlbmd0aA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5lbnZfc3RhdGUucnVubmluZ19jb3VudF9lcGlzb2RlX3JldHVybg==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5rZXk=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLmFjdGlvbl9oZWFkLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLmFjdGlvbl9oZWFkLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLmNyaXRpY19oZWFkLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLmNyaXRpY19oZWFkLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8xLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8yLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubXUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8yLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLmFjdGlvbl9oZWFkLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLmFjdGlvbl9oZWFkLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLmNyaXRpY19oZWFkLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLmNyaXRpY19oZWFkLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8xLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8xLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8yLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmFjX29wdF9zdGF0ZS4xLjAubnUucGFyYW1zLnNoYXJlZF90b3Jzby5EZW5zZV8yLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmR1YWxfb3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmR1YWxfb3B0X3N0YXRlLjEuMC5tdS5sb2dfYWxwaGE=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmR1YWxfb3B0X3N0YXRlLjEuMC5tdS5sb2dfdGVtcGVyYXR1cmU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmR1YWxfb3B0X3N0YXRlLjEuMC5udS5sb2dfYWxwaGE=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5vcHRfc3RhdGVzLmR1YWxfb3B0X3N0YXRlLjEuMC5udS5sb2dfdGVtcGVyYXR1cmU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5hY3Rpb25faGVhZC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5hY3Rpb25faGVhZC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5jcml0aWNfaGVhZC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5jcml0aWNfaGVhZC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5zaGFyZWRfdG9yc28uRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5zaGFyZWRfdG9yc28uRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5zaGFyZWRfdG9yc28uRGVuc2VfMS5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5zaGFyZWRfdG9yc28uRGVuc2VfMS5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5zaGFyZWRfdG9yc28uRGVuc2VfMi5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuYWNfcGFyYW1zLnBhcmFtcy5zaGFyZWRfdG9yc28uRGVuc2VfMi5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuZHVhbF9wYXJhbXMubG9nX2FscGhh":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMuZHVhbF9wYXJhbXMubG9nX3RlbXBlcmF0dXJl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuY3JpdGljX2hlYWQuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuY3JpdGljX2hlYWQuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuYWN0aW9uX2hlYWQuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuYWN0aW9uX2hlYWQuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuc2hhcmVkX3RvcnNvLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuc2hhcmVkX3RvcnNvLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuc2hhcmVkX3RvcnNvLkRlbnNlXzEuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuc2hhcmVkX3RvcnNvLkRlbnNlXzEua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuc2hhcmVkX3RvcnNvLkRlbnNlXzIuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}","bGVhcm5lcl9zdGF0ZS5wYXJhbXMudGFyZ2V0X2FjX3BhcmFtcy5wYXJhbXMuc2hhcmVkX3RvcnNvLkRlbnNlXzIua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"TPU_0(process=0,(0,0,0,0))\"}"}

checkpoints/94789632/default/ocdbt.process_0/d/764408dd0353b32b2d8425bfb4cf4b99

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

checkpoints/94789632/metrics/metrics

@@ -0,0 +1 @@
+{"episode_return": 500.0}