diff --git a/train.py b/train.py
index 01429c0e9b39..257be065f641 100644
--- a/train.py
+++ b/train.py
@@ -6,6 +6,7 @@
 
 import argparse
 import logging
+import math
 import os
 import random
 import sys
@@ -15,7 +16,6 @@
 from pathlib import Path
 from threading import Thread
 
-import math
 import numpy as np
 import torch.distributed as dist
 import torch.nn as nn