Spaces:
Runtime error
Runtime error
File size: 526 Bytes
fc16538 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# TRI-VIDAR - Copyright 2022 Toyota Research Institute. All rights reserved.
import os
import fire
import torch
from vidar.core.trainer import Trainer
from vidar.core.wrapper import Wrapper
from vidar.utils.config import read_config
def train(cfg, **kwargs):
os.environ['DIST_MODE'] = 'gpu' if torch.cuda.is_available() else 'cpu'
cfg = read_config(cfg, **kwargs)
wrapper = Wrapper(cfg, verbose=True)
trainer = Trainer(cfg)
trainer.learn(wrapper)
if __name__ == '__main__':
fire.Fire(train)
|