Upload folder using huggingface_hub

.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
+39911424/default/ocdbt.process_0/d/3f4d40e8fed8c706a35861167e7ba406 filter=lfs diff=lfs merge=lfs -text

39911424/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"init_timestamp_nsecs": 1729019372589943585, "commit_timestamp_nsecs": 1729019372732271122}

39911424/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}

39911424/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))\"}"}

39911424/default/ocdbt.process_0/d/3f4d40e8fed8c706a35861167e7ba406

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

39911424/metrics/metrics

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

metadata/_CHECKPOINT_METADATA

@@ -0,0 +1 @@
+{"init_timestamp_nsecs": 1729019272037315713, "commit_timestamp_nsecs": 1729019272039572093}

metadata/metadata

@@ -0,0 +1 @@
+{"checkpointer_version": 1.0, "arch": {"absolute_metric": true, "evaluation_greedy": false, "num_envs": 256, "num_eval_episodes": 128, "num_evaluation": 20, "num_updates": 4650.0, "num_updates_per_eval": 232.0, "seed": 0, "total_num_envs": 1024, "total_timesteps": 100000000.0, "update_batch_size": 1}, "env": {"env_name": "gymnax", "eval_metric": "episode_return", "kwargs": {}, "scenario": {"name": "CartPole-v1", "task_name": "cartpole"}}, "logger": {"base_exp_path": "results", "checkpointing": {"load_args": {"checkpoint_uid": ""}, "load_model": false, "save_args": {"checkpoint_uid": null, "hf_repo_name": "InstaDeepAI/spo-benchmark-brax-ant", "keep_period": null, "max_to_keep": 1, "save_interval_steps": 1}, "save_model": true}, "kwargs": {"detailed_neptune_logging": false, "json_path": null, "neptune_project": "InstaDeep/flashbax", "neptune_tag": [], "upload_json_data": false}, "use_console": true, "use_json": false, "use_neptune": false, "use_tb": false}, "network": {"joint_network": {"action_head": {"_target_": "spo.networks.heads.CategoricalHead"}, "critic_head": {"_target_": "spo.networks.heads.ScalarCriticHead"}, "shared_torso": {"_target_": "spo.networks.torso.JointMLPTorso", "activation": "silu", "critic_layer_sizes": [256], "policy_layer_sizes": [256], "shared_layer_sizes": [256], "use_layer_norm": false}}}, "num_devices": 4, "system": {"action_dim": 2, "batch_size": 16, "buffer_size": 16384, "decay_learning_rates": false, "dirichlet_alpha": 0.03, "dual_lr": 0.0003, "ent_coef": 0.0, "epochs": 16, "epsilon": 0.5, "epsilon_policy": 0.001, "gae_lambda": 0.95, "gamma": 0.99, "init_log_alpha": 0.5, "init_log_temperature": 0.5, "lr": 0.0003, "max_depth": 4, "max_grad_norm": 0.5, "num_particles": 16, "period": 1, "resampling_period": 4, "rollout_length": 21, "root_exploration_fraction": 0.25, "sample_sequence_length": 17, "search_gae_lambda": 1.0, "search_gamma": 1.0, "system_name": "ff_smx", "tau": 0.005, "total_batch_size": 64, "total_buffer_size": 65536, "vf_coef": 1.0, "warmup_steps": 0}}