Browse Source
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
2 additions and
1 deletions
-
clmr_magnatagatune.py
|
|
@ -18,7 +18,6 @@ import logging |
|
|
from pathlib import Path |
|
|
from pathlib import Path |
|
|
from typing import Union |
|
|
from typing import Union |
|
|
|
|
|
|
|
|
import subprocess |
|
|
|
|
|
import torchaudio |
|
|
import torchaudio |
|
|
import torch |
|
|
import torch |
|
|
import numpy |
|
|
import numpy |
|
|
@ -31,6 +30,8 @@ sys.path.append(str(Path(__file__).parent)) |
|
|
from utils.checkpoint import load_encoder_checkpoint |
|
|
from utils.checkpoint import load_encoder_checkpoint |
|
|
from models.sample_cnn import SampleCNN |
|
|
from models.sample_cnn import SampleCNN |
|
|
|
|
|
|
|
|
|
|
|
log = logging.getLogger() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@register(output_schema=['vec']) |
|
|
@register(output_schema=['vec']) |
|
|
class ClmrMagnatagatune(NNOperator): |
|
|
class ClmrMagnatagatune(NNOperator): |
|
|
|
