diff --git a/hydragnn/models/create.py b/hydragnn/models/create.py index 8cb08a655..cff9d8784 100644 --- a/hydragnn/models/create.py +++ b/hydragnn/models/create.py @@ -99,7 +99,7 @@ def create_model( num_filters: int = None, radius: float = None, equivariance: bool = False, - conv_checkopinting: bool = False, + conv_checkpointing: bool = False, verbosity: int = 0, use_gpu: bool = True, ): @@ -354,7 +354,7 @@ def create_model( else: raise ValueError("Unknown model_type: {0}".format(model_type)) - if conv_checkopinting: + if conv_checkpointing: model.enable_conv_checkpointing() timer.stop()