Spaces:
Runtime error
Runtime error
Hugo Flores Garcia
commited on
Commit
·
980f2e7
1
Parent(s):
a876330
drop tensorboardX
Browse files- scripts/exp/train.py +1 -1
- setup.py +0 -1
scripts/exp/train.py
CHANGED
@@ -14,7 +14,7 @@ from audiotools.data import transforms
|
|
14 |
from einops import rearrange
|
15 |
from rich import pretty
|
16 |
from rich.traceback import install
|
17 |
-
from
|
18 |
|
19 |
import vampnet
|
20 |
from vampnet.modules.transformer import VampNet
|
|
|
14 |
from einops import rearrange
|
15 |
from rich import pretty
|
16 |
from rich.traceback import install
|
17 |
+
from torch.utils.tensorboard import SummaryWriter
|
18 |
|
19 |
import vampnet
|
20 |
from vampnet.modules.transformer import VampNet
|
setup.py
CHANGED
@@ -33,7 +33,6 @@ setup(
|
|
33 |
"lac @ git+https://github.com/hugofloresgarcia/lac.git",
|
34 |
"descript-audiotools @ git+https://github.com/descriptinc/audiotools.git@0.7.2",
|
35 |
"gradio",
|
36 |
-
"tensorboardX",
|
37 |
"loralib",
|
38 |
"torch_pitch_shift",
|
39 |
"madmom",
|
|
|
33 |
"lac @ git+https://github.com/hugofloresgarcia/lac.git",
|
34 |
"descript-audiotools @ git+https://github.com/descriptinc/audiotools.git@0.7.2",
|
35 |
"gradio",
|
|
|
36 |
"loralib",
|
37 |
"torch_pitch_shift",
|
38 |
"madmom",
|