forked from kaist-avelab/K-Lane
-
Notifications
You must be signed in to change notification settings - Fork 0
/
train_gpu_0.py
41 lines (33 loc) · 1.25 KB
/
train_gpu_0.py
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
'''
* Copyright (c) AVELab, KAIST. All rights reserved.
* author: Donghee Paek & Kevin Tirta Wijaya, AVELab, KAIST
* e-mail: donghee.paek@kaist.ac.kr, kevin.tirta@kaist.ac.kr
'''
import os
GPUS_EN = '0'
os.environ["CUDA_VISIBLE_DEVICES"] = GPUS_EN
import torch
import torch.backends.cudnn as cudnn
import time
import shutil
time_now = time.localtime()
time_log = '%04d-%02d-%02d-%02d-%02d-%02d' % (time_now.tm_year, time_now.tm_mon, time_now.tm_mday, time_now.tm_hour, time_now.tm_min, time_now.tm_sec)
from baseline.utils.config import Config
from baseline.engine.runner import Runner
def main():
path_config = './configs/Proj28_GFC-T3_RowRef_82_73.py'
path_split = path_config.split('/')
cfg = Config.fromfile(path_config)
cfg.log_dir = cfg.log_dir + '/' + time_log
cfg.time_log = time_log
os.makedirs(cfg.log_dir, exist_ok=True)
shutil.copyfile(path_config, cfg.log_dir + '/' + path_split[-2]
+ '_' + path_split[-1].split('.')[0] + '.txt')
cfg.work_dirs = cfg.log_dir + '/' + cfg.dataset.train.type
cfg.gpus = len(GPUS_EN.split(','))
cudnn.benchmark = True
runner = Runner(cfg)
runner.train()
# runner.train_small(train_batch=2, valid_samples=40)
if __name__ == '__main__':
main()