Commit fb5c9ff4 authored by Verena Praher's avatar Verena Praher
Browse files

add paths for Verena

parent b04cc45e
import os import os
import time import time
import hashlib import hashlib
import getpass
import datetime import datetime
import torch import torch
import torch.nn as nn import torch.nn as nn
...@@ -23,8 +24,14 @@ MAX_LENGTH = 10000 ...@@ -23,8 +24,14 @@ MAX_LENGTH = 10000
# CONFIG ======================= # CONFIG =======================
# paths: # paths:
PATH_PROJECT_ROOT = os.path.dirname(os.path.realpath(__file__))
PATH_RESULTS = os.path.join(PATH_PROJECT_ROOT, 'results')
hostname = os.uname()[1] hostname = os.uname()[1]
if hostname in ['rechenknecht3.cp.jku.at']: username = getpass.getuser()
if hostname in ['rechenknecht3.cp.jku.at', 'rechenknecht2.cp.jku.at']:
plt.switch_backend('agg') plt.switch_backend('agg')
PATH_DATA_ROOT = '/media/rk3/shared/datasets/MTG-Jamendo' PATH_DATA_ROOT = '/media/rk3/shared/datasets/MTG-Jamendo'
USE_GPU = True USE_GPU = True
...@@ -32,6 +39,9 @@ elif hostname == 'hermine': # PC verena ...@@ -32,6 +39,9 @@ elif hostname == 'hermine': # PC verena
plt.switch_backend('agg') plt.switch_backend('agg')
PATH_DATA_ROOT = '/media/verena/SAMSUNG/Data/MTG-Jamendo' PATH_DATA_ROOT = '/media/verena/SAMSUNG/Data/MTG-Jamendo'
USE_GPU = True USE_GPU = True
elif hostname == 'verena-830g5': # Laptop Verena
PATH_DATA_ROOT = '/media/verena/SAMSUNG/Data/MTG-Jamendo'
USE_GPU = False
elif hostname == 'shreyan-HP': # Laptop Shreyan elif hostname == 'shreyan-HP': # Laptop Shreyan
PATH_DATA_ROOT = '/home/shreyan/mounts/home@rk3/shared/datasets/MTG-Jamendo' PATH_DATA_ROOT = '/home/shreyan/mounts/home@rk3/shared/datasets/MTG-Jamendo'
USE_GPU = False USE_GPU = False
...@@ -39,15 +49,20 @@ else: ...@@ -39,15 +49,20 @@ else:
PATH_DATA_ROOT = '/mnt/2tb/datasets/MTG-Jamendo' PATH_DATA_ROOT = '/mnt/2tb/datasets/MTG-Jamendo'
USE_GPU = False USE_GPU = False
PATH_PROJECT_ROOT = os.path.dirname(os.path.realpath(__file__)) if username == 'verena':
PATH_RESULTS = '/home/verena/experiments/moodwalk/'
PATH_AUDIO = os.path.join(PATH_DATA_ROOT, 'MTG-Jamendo_audio') PATH_AUDIO = os.path.join(PATH_DATA_ROOT, 'MTG-Jamendo_audio')
PATH_ANNOTATIONS = os.path.join(PATH_DATA_ROOT, 'MTG-Jamendo_annotations') PATH_ANNOTATIONS = os.path.join(PATH_DATA_ROOT, 'MTG-Jamendo_annotations')
PATH_MELSPEC_DOWNLOADED = os.path.join(PATH_DATA_ROOT, 'MTG-Jamendo_melspec_downloaded') PATH_MELSPEC_DOWNLOADED = os.path.join(PATH_DATA_ROOT, 'MTG-Jamendo_melspec_downloaded')
PATH_MELSPEC_DOWNLOADED_FRAMED = os.path.join(PATH_MELSPEC_DOWNLOADED, 'framed') PATH_MELSPEC_DOWNLOADED_FRAMED = os.path.join(PATH_MELSPEC_DOWNLOADED, 'framed')
PATH_MELSPEC_DOWNLOADED_HDF5 = os.path.join(PATH_DATA_ROOT, 'HDF5Cache_spectrograms') PATH_MELSPEC_DOWNLOADED_HDF5 = os.path.join(PATH_DATA_ROOT, 'HDF5Cache_spectrograms')
PATH_RESULTS = os.path.join(PATH_PROJECT_ROOT, 'results')
TRAINED_MODELS_PATH = '' TRAINED_MODELS_PATH = ''
# run name # run name
def make_run_name(suffix=''): def make_run_name(suffix=''):
assert ' ' not in suffix assert ' ' not in suffix
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment