Skip to content

Commit

Permalink
Fix a Typo (#292)
Browse files Browse the repository at this point in the history
  • Loading branch information
RylieWeaver authored Oct 8, 2024
1 parent e532659 commit f8f7893
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions hydragnn/models/create.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
):
Expand Down Expand Up @@ -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()
Expand Down

0 comments on commit f8f7893

Please sign in to comment.