diff --git a/references/script.py b/references/script.py
index 7b423d9037..117a01d4b1 100644
--- a/references/script.py
+++ b/references/script.py
@@ -5,7 +5,6 @@
 import torch.nn as nn
 import torch.nn.functional as F
 import numpy as np
-import math
 from typing import Dict, List, Tuple, Optional
 from collections import OrderedDict