diff --git a/slsa_for_models/pytorch_cifar10.py b/slsa_for_models/pytorch_cifar10.py index 66893a71..0d3ccee5 100644 --- a/slsa_for_models/pytorch_cifar10.py +++ b/slsa_for_models/pytorch_cifar10.py @@ -15,6 +15,7 @@ import torch import torch.nn as nn + # We will do a lazy import for these 5 modules, exploiting Python's symbol # resolution. The lazy import is needed to make sure we only import PyTorch # libraries only if we want to train a PyTorch model. @@ -111,8 +112,6 @@ def create_model(): Returns the model. """ - - return MyModel()