import os
import sys
__root = os.path.abspath(
os.path.dirname(os.path.abspath(__file__)) + (os.sep + '..') *
(len(os.path.dirname(os.path.abspath(__file__)).split(os.sep)) - os.path.
dirname(os.path.abspath(__file__)).split(os.sep).index('VeXtract') - 1)
) + os.sep
sys.path.append(__root)
_ROOT = __root
from helper import logger
log = logger.Logger(__name__)
from core.scene.scene import Scene
from core.common.segment import Segment
SCENE = Scene()
SCENE.add_segment(Segment((2, 15.5)))
SCENE.add_segment(Segment((25, 30)))
SCENE2 = Scene()
SCENE2.add_segment(Segment((11, 15.5)))
SCENE2.add_segment(Segment((25, 27)))
SCENE_LIST = [SCENE, SCENE2]
def test_scene_str():
assert str(SCENE) == "(2, 30)"
model=hiagm,
config=config)
# get epoch trainer
trainer = Trainer(model=hiagm,
criterion=criterion,
optimizer=optimize,
vocab=corpus_vocab,
config=config)
hiagm.eval()
# set origin log
trainer.eval(test_loader, -1, 'TEST')
return
if __name__ == "__main__":
configs = Configure(config_json_file=sys.argv[1])
if configs.train.device_setting.device == 'cuda':
os.system('CUDA_VISIBLE_DEVICES=' +
str(configs.train.device_setting.visible_device_list))
else:
os.system("CUDA_VISIBLE_DEVICES=''")
torch.manual_seed(2019)
torch.cuda.manual_seed(2019)
logger.Logger(configs)
if not os.path.isdir(configs.train.checkpoint.dir):
os.mkdir(configs.train.checkpoint.dir)
evaluate(configs)
