Skip to content

P
PaddleSlim

PaddlePaddle / PaddleSlim
大约 1 年 前同步成功

通知 51
Star 1434
Fork 344
P
PaddleSlim
未验证 提交 db7329bc 编写于 作者: C ceci3 提交者： GitHub
浏览文件
操作

change default class dims (#263)

上级 f1b2e3b7
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
demo/nas/rl_nas_mobilenetv2.py
浏览文件 @ db7329bc
......@@ -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,
......
demo/nas/sa_nas_mobilenetv2.py
浏览文件 @ db7329bc
......@@ -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,
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或