diff --git a/main_custom.py b/main_custom.py
index a3faecb8ab69959892edc760db929ccfe57300bf..7048332c805a7e88a84b13704abe9d8b52e78b17 100644
--- a/main_custom.py
+++ b/main_custom.py
@@ -1,5 +1,4 @@
 import os
-os.environ['CUDA_LAUNCH_BLOCKING'] = "1"
 import pandas as pd
 import torch
 import torch.optim as optim