Prune stale checkpoint step folders batch 1

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/10000/_CHECKPOINT_METADATA

@@ -1 +0,0 @@
-{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1778300662857490720, "commit_timestamp_nsecs": 1778300665865494342, "custom_metadata": {}}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/10000/default/_sharding

@@ -1 +0,0 @@
-{"Y3JpdGljLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm11LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm51LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5wYXJhbXMubG9nX2FscGhh":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3Jld2FyZF9zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3RhcmdldF9yZXdhcmRfc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cmV3YXJkX25vcm1fc3Rk":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/10000/default/array_metadatas/process_0

@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "critic.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.params.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.mu.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.nu.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "reward_norm_std", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "mrq_reward_scale", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "mrq_target_reward_scale", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}]}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/10000/default/ocdbt.process_0/d/5d5322086f519d9ffd060148195c6ead

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/10000/default/ocdbt.process_0/d/dd04906a42542bbff05278fbc91240ea

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/10000/default/ocdbt.process_0/d/e4e6072233f4d98611af98f3b3c02383

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/100000/_CHECKPOINT_METADATA

@@ -1 +0,0 @@
-{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1778312774728860264, "commit_timestamp_nsecs": 1778312777712330539, "custom_metadata": {}}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/100000/default/_sharding

@@ -1 +0,0 @@
-{"Y3JpdGljLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm11LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm51LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5wYXJhbXMubG9nX2FscGhh":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3Jld2FyZF9zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3RhcmdldF9yZXdhcmRfc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cmV3YXJkX25vcm1fc3Rk":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/100000/default/array_metadatas/process_0

@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "critic.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.params.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.mu.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.nu.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "reward_norm_std", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "mrq_reward_scale", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "mrq_target_reward_scale", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}]}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/100000/default/ocdbt.process_0/d/528b4704d355f6ca05b1bfdab18c56af

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/100000/default/ocdbt.process_0/d/5ebb1bf3c920a89cd3e128393e3b6a0a

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/100000/default/ocdbt.process_0/d/dee862e6c2f9d0197852fa260d2f37c2

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/110000/_CHECKPOINT_METADATA

@@ -1 +0,0 @@
-{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1778313660574949273, "commit_timestamp_nsecs": 1778313663275333091, "custom_metadata": {}}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/110000/default/_sharding

@@ -1 +0,0 @@
-{"Y3JpdGljLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm11LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm51LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5wYXJhbXMubG9nX2FscGhh":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3Jld2FyZF9zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3RhcmdldF9yZXdhcmRfc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cmV3YXJkX25vcm1fc3Rk":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/110000/default/array_metadatas/process_0

@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "critic.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.params.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.mu.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.nu.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "reward_norm_std", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "mrq_reward_scale", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "mrq_target_reward_scale", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}]}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/110000/default/ocdbt.process_0/d/765f6e45510523f1b24a99edff08aec6

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/110000/default/ocdbt.process_0/d/a6db268020a5a59953254542130075ff

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/110000/default/ocdbt.process_0/d/f4e7753b68af5ab70f7653d94869b706

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/120000/_CHECKPOINT_METADATA

@@ -1 +0,0 @@
-{"item_handlers": {"default": "orbax.checkpoint._src.handlers.standard_checkpoint_handler.StandardCheckpointHandler"}, "metrics": {}, "performance_metrics": {}, "init_timestamp_nsecs": 1778314703114674823, "commit_timestamp_nsecs": 1778314706171744332, "custom_metadata": {}}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/120000/default/_sharding

@@ -1 +0,0 @@
-{"Y3JpdGljLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnBhcmFtcy5wYXJhbXMuQmlubmVkUmVncmVzc2lvbl8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5wYXJhbXMucGFyYW1zLkJpbm5lZFJlZ3Jlc3Npb25fMS5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","Y3JpdGljX3RhcmdldC5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5jb3VudA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5tdS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iub3B0X3N0YXRlLjEuMC5udS5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3Iuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","YWxwaGFfYWN0b3IucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8yLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm11LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLm51LmxvZ19hbHBoYQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5wYXJhbXMubG9nX2FscGhh":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW50cm9weS5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm11LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLm51LnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5vcHRfc3RhdGUuMS4wLmNvdW50":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5wYXJhbXMucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlci5zdGVw":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQuc3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8wLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkRlbnNlXzAua2VybmVs":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","ZW5jb2Rlcl90YXJnZXQucGFyYW1zLnBhcmFtcy5NTFBfMC5MYXllcl8xLkxheWVyTm9ybV8wLnNjYWxl":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAuY291bnQ=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubXUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8wLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8xLkxheWVyXzIuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzAuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuYmlhcw==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzEuTGF5ZXJOb3JtXzAuc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLm9wdF9zdGF0ZS4xLjAubnUucGFyYW1zLk1MUF8yLkxheWVyXzIuRGVuc2VfMC5rZXJuZWw=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzAuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzEuTGF5ZXJfMi5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMC5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5iaWFz":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMS5MYXllck5vcm1fMC5zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmJpYXM=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnBhcmFtcy5wYXJhbXMuTUxQXzIuTGF5ZXJfMi5EZW5zZV8wLmtlcm5lbA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bGF0ZW50X21vZGVsLnN0ZXA=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3Jld2FyZF9zY2FsZQ==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","bXJxX3RhcmdldF9yZXdhcmRfc2NhbGU=":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","c3RlcA==":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}","cmV3YXJkX25vcm1fc3Rk":"{\"sharding_type\": \"SingleDeviceSharding\", \"device_str\": \"cuda:0\"}"}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/120000/default/array_metadatas/process_0

@@ -1 +0,0 @@
-{"array_metadatas": [{"array_metadata": {"param_name": "critic.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.params.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.mu.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic.opt_state.1.0.nu.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_2.Dense_0.bias", "write_shape": [10, 751], "chunk_shape": [10, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "critic_target.params.params.BinnedRegression_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 751], "chunk_shape": [10, 512, 751], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 27, 512], "chunk_shape": [10, 27, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "encoder_target.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.params.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.mu.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_1.Layer_2.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.Dense_0.kernel", "write_shape": [10, 520, 512], "chunk_shape": [10, 520, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_1.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_2.Dense_0.bias", "write_shape": [10, 1], "chunk_shape": [10, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "latent_model.opt_state.1.0.nu.params.MLP_2.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 1], "chunk_shape": [10, 512, 1], "ext_metadata": null}}, {"array_metadata": {"param_name": "step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.params.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.mu.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "entropy.opt_state.1.0.nu.log_alpha", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.step", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.params.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.count", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.mu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_0.LayerNorm_0.scale", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.bias", "write_shape": [10, 512], "chunk_shape": [10, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_1.Dense_0.kernel", "write_shape": [10, 512, 512], "chunk_shape": [10, 512, 512], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.bias", "write_shape": [10, 16], "chunk_shape": [10, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "alpha_actor.opt_state.1.0.nu.params.MLP_0.Layer_2.Dense_0.kernel", "write_shape": [10, 512, 16], "chunk_shape": [10, 512, 16], "ext_metadata": null}}, {"array_metadata": {"param_name": "reward_norm_std", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "mrq_reward_scale", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}, {"array_metadata": {"param_name": "mrq_target_reward_scale", "write_shape": [10], "chunk_shape": [10], "ext_metadata": null}}]}

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/120000/default/ocdbt.process_0/d/3ff8fbdcb1bb2e8905fdb0243f8ae77a

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

Ant/AE_MAD_SAC_kappa0p05_maxalpha6/10seed/checkpoint/120000/default/ocdbt.process_0/d/851a4d60881f5101c4b86d1e7a19a531

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