Spaces:
Sleeping
Sleeping
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",
|