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