nvalchemi.training.register_type_serializer#

nvalchemi.training.register_type_serializer(type_, serialize, deserialize)[source]#

Register JSON (de)serializers for a custom type.

Parameters:
  • type – The Python type to register, for example torch.dtype.

  • serialize (Callable[[Any], Any]) – Callable converting a type_ instance to a JSON-safe value.

  • deserialize (Callable[[Any], Any]) – Callable converting the JSON-safe value back into a type_ instance.

  • type_ (type)

Return type:

None