main.py

import os
import resource

from helpers import *
from Trainer import Trainer

rlimit = resource.getrlimit(resource.RLIMIT_NOFILE)
resource.setrlimit(
    resource.RLIMIT_NOFILE, (65536, rlimit[1])
)

if __name__ == '__main__':
    opt = __import__('options')
    os.environ['CUDA_VISIBLE_DEVICES'] = opt.gpu
else:
    import options as opt


if __name__ == '__main__':
    print("Loading options...")
    trainer = Trainer()

    if hasattr(opt, 'weights'):
        trainer.load_weights(opt.weights)

    torch.manual_seed(opt.random_seed)
    torch.cuda.manual_seed_all(opt.random_seed)
    trainer.train()