diff --git a/prototorch/models/abstract.py b/prototorch/models/abstract.py
index e21b4d7..529adf6 100644
--- a/prototorch/models/abstract.py
+++ b/prototorch/models/abstract.py
@@ -1,5 +1,4 @@
 import pytorch_lightning as pl
-import torch
 from torch.optim.lr_scheduler import ExponentialLR