jaxrts.experimental.SiiNN.NNSiiModel.__init__

NNSiiModel.__init__(checkpoint_dir: Path)[source]

Initialize the NNSiiModel.

Parameters:

checkpoint_dir (Path) – The Path to the checkpoint directory.