diff --git a/examples/models/nn.py b/examples/models/nn.py
index df08d72..64a3deb 100644
--- a/examples/models/nn.py
+++ b/examples/models/nn.py
@@ -14,7 +14,6 @@
 import torch
 import numpy as np
 from torch import nn
-from torch.nn import functional as F
 
 
 class LeNet5(nn.Module):