if sys.version_info[0] >= 3:
sys.stdout = os.fdopen(sys.stdout.fileno(), "w")
else:
sys.stdout = os.fdopen(sys.stdout.fileno(), "w", 0)
class NetFactory(object):
After Change
if util.has_bad_inputs(args):
sys.exit(-1)
if not (args.cuda_devices == """"):
os.environ["CUDA_VISIBLE_DEVICES"] = args.cuda_devices
print("set CUDA_VISIBLE_DEVICES env to {}".format(args.cuda_devices))
net_class = NetFactory.create(args.net_name)
