| 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() | |