mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update

pull/17/head
KaiyangZhou 2018-03-11 21:42:57 +00:00
parent b0af5b0914
commit b9150adc45
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
train_img_model_xent.py
View File

@ -50,7 +50,7 @@ parser.add_argument('--print-freq', type=int, default=5, help="print frequency")
parser.add_argument('--seed', type=int, default=1, help="manual seed")
parser.add_argument('--resume', type=str, default='', metavar='PATH')
parser.add_argument('--evaluate', action='store_true', help="evaluation only")
parser.add_argument('--eval-step', type=int, default=50, help="every N epochs to test")
parser.add_argument('--eval-step', type=int, default=50, help="run evaluation for every N epochs")
parser.add_argument('--save-dir', type=str, default='log')
parser.add_argument('--use-cpu', action='store_true', help="use cpu")
parser.add_argument('--gpu-devices', default='0', type=str, help='gpu device ids for CUDA_VISIBLE_DEVICES')
@ -64,7 +64,7 @@ def main():
use_gpu = torch.cuda.is_available()
if args.use_cpu: use_gpu = False
print("==========\nArgs:{}\n==========".format(locals().copy()))
print("==========\nArgs:{}\n==========".format(args))
if use_gpu:
print("Currently using GPU")