diff --git a/u2net_train.py b/u2net_train.py
index 4c2dcaa..40395e9 100644
--- a/u2net_train.py
+++ b/u2net_train.py
@@ -160,5 +160,3 @@ for epoch in range(0, epoch_num):
             net.train()  # resume train
             ite_num4val = 0
 
-if __name__ == "__main__":
-    main()