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