maximeseince's picture
Retrain context_model with SumContextEncoder-based network (time+label folded into one context vector)
27bbec5 verified
Raw
History Blame Contribute Delete
5.78 kB
{"tree_metadata": {"('network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'context_encoders', 'label', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'context_encoders', 'label', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "context_encoders", "key_type": 2}, {"key": "label", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'context_encoder', 'time_encoder', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "context_encoder", "key_type": 2}, {"key": "time_encoder", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'coordinates', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'heads', 'coordinates', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "heads", "key_type": 2}, {"key": "coordinates", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '0', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "0", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'bias', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "bias", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}, "('network', 'trunk', 'layers', '2', 'kernel', 'value')": {"key_metadata": [{"key": "network", "key_type": 2}, {"key": "trunk", "key_type": 2}, {"key": "layers", "key_type": 2}, {"key": "2", "key_type": 2}, {"key": "kernel", "key_type": 2}, {"key": "value", "key_type": 2}], "value_metadata": {"value_type": "np.ndarray", "skip_deserialize": false}}}, "use_ocdbt": true, "use_zarr3": false, "store_array_data_equal_to_fill_value": true, "custom_metadata": null}