From db7329bcc8e997d49b67e8836ff85346d68d6485 Mon Sep 17 00:00:00 2001 From: ceci3 Date: Fri, 8 May 2020 15:41:24 +0800 Subject: [PATCH] change default class dims (#263) --- demo/nas/rl_nas_mobilenetv2.py | 2 +- demo/nas/sa_nas_mobilenetv2.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/demo/nas/rl_nas_mobilenetv2.py b/demo/nas/rl_nas_mobilenetv2.py index 2fda5831..abf23fb9 100644 --- a/demo/nas/rl_nas_mobilenetv2.py +++ b/demo/nas/rl_nas_mobilenetv2.py @@ -182,7 +182,7 @@ if __name__ == '__main__': parser.add_argument( '--batch_size', type=int, default=256, help='batch size.') parser.add_argument( - '--class_dim', type=int, default=1000, help='classify number.') + '--class_dim', type=int, default=10, help='classify number.') parser.add_argument( '--data', type=str, diff --git a/demo/nas/sa_nas_mobilenetv2.py b/demo/nas/sa_nas_mobilenetv2.py index f1ebd81f..9ef26bc5 100644 --- a/demo/nas/sa_nas_mobilenetv2.py +++ b/demo/nas/sa_nas_mobilenetv2.py @@ -271,7 +271,7 @@ if __name__ == '__main__': parser.add_argument( '--batch_size', type=int, default=256, help='batch size.') parser.add_argument( - '--class_dim', type=int, default=1000, help='classify number.') + '--class_dim', type=int, default=10, help='classify number.') parser.add_argument( '--data', type=str, -- GitLab