StyleFeatureEditor

1
import sys
2
import torch
3

4
sys.path =  ['.'] + sys.path
5

6
from arguments import inference_arguments
7
from runners.inference_runners import inference_runner_registry
8
from utils.common_utils import printer, setup_seed
9

10

11
def run_inference(config):
12
    inference_runner = inference_runner_registry[config.inference.inference_runner](
13
        config
14
    )
15
    inference_runner.setup()
16
    inference_runner.run()
17

18

19
if __name__ == "__main__":
20
    config = inference_arguments.load_config()
21
    setup_seed(config.exp.seed)
22

23
    printer(config)
24

25
    run_inference(config)
26