File size: 287 Bytes
f7f604d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import torch.cuda as cuda
from utils.misc import *
from run import *
if __name__ == "__main__":
args = parse_args()
opt = load_config(args.config)
train(opt, args)
cuda.empty_cache()
if args.local_rank <= 0:
test(opt, args)
evaluate(opt, args)
|