From 8e58a7c65ebaa88383b2091c77d088f3e0421421 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 21 Oct 2024 13:44:47 +0200
Subject: [PATCH] datasets

---
 main_custom.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main_custom.py b/main_custom.py
index b130650..a3faecb 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
-- 
GitLab