Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Shreyan Chowdhury
moodwalk
Commits
fb5c9ff4
Commit
fb5c9ff4
authored
Sep 07, 2019
by
Verena Praher
Browse files
add paths for Verena
parent
b04cc45e
Changes
1
Hide whitespace changes
Inline
Side-by-side
utils.py
View file @
fb5c9ff4
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
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment