Skip to content
Snippets Groups Projects
Commit 1920d1a6 authored by Maria Belen Guaranda Cabezas's avatar Maria Belen Guaranda Cabezas
Browse files

hot fixes so it can run in the cluster

parent ab8bdb67
No related branches found
No related tags found
No related merge requests found
......@@ -21,7 +21,6 @@ from torch.multiprocessing import spawn
import os
import sys
SOURCE_DIR = os.path.abspath(os.path.join(os.path.dirname( __file__ ), '../..', 'src'))
print(SOURCE_DIR)
sys.path.append(SOURCE_DIR)
from learner import train, train_distributed
......
......@@ -23,6 +23,12 @@ from concurrent.futures import ProcessPoolExecutor
from glob import glob
from tqdm import tqdm
# Needed to import from src/ in general
import os
import sys
SOURCE_DIR = os.path.abspath(os.path.join(os.path.dirname( __file__ ), '../..', 'src'))
sys.path.append(SOURCE_DIR)
from diffwave.params import params
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment