mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-13 23:53:02 +00:00
Brandon Thomas
16 lines
351 B
Python
16 lines
351 B
Python
import os
|
|
import sys
|
|
import copy
|
|
import json
|
|
import math
|
|
import random
|
|
import logging
|
|
import itertools
|
|
import numpy as np
|
|
|
|
from utils.config import Config
|
|
from utils.registry_class import INFER_ENGINE
|
|
|
|
if __name__ == '__main__':
|
|
cfg_update = Config(load=True)
|
|
INFER_ENGINE.build(dict(type=cfg_update.TASK_TYPE), cfg_update=cfg_update.cfg_dict) |