import argparse
def parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--gpu', type=str, dest='gpu_ids')
parser.add_argument('--test_epoch', type=str, dest='test_epoch')
args = parser.parse_args()
# test gpus
if not args.gpu_ids:
assert 0, print("Please set proper gpu ids")
if '-' in args.gpu_ids:
gpus = args.gpu_ids.split('-')
gpus[0] = int(gpus[0])
gpus[1] = int(gpus[1]) + 1
args.gpu_ids = ','.join(map(lambda x: str(x), list(range(*gpus))))
assert args.test_epoch, 'Test epoch is required.'
return args
parse_args()
輸出為：
Please set proper gpu ids
Traceback (most recent call last):
File "C:\Users\user\Desktop\tmp\test.py", line 22, in <module>
parse_args()
File "C:\Users\user\Desktop\tmp\test.py", line 12, in parse_args
assert 0, print("Please set proper gpu ids")
AssertionError: None
哭了 為什么啊
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/241547.html
上一篇：下載問題