未验证 提交 db7329bc 编写于 作者: C ceci3 提交者: GitHub

change default class dims (#263)

上级 f1b2e3b7
...@@ -182,7 +182,7 @@ if __name__ == '__main__': ...@@ -182,7 +182,7 @@ if __name__ == '__main__':
parser.add_argument( parser.add_argument(
'--batch_size', type=int, default=256, help='batch size.') '--batch_size', type=int, default=256, help='batch size.')
parser.add_argument( parser.add_argument(
'--class_dim', type=int, default=1000, help='classify number.') '--class_dim', type=int, default=10, help='classify number.')
parser.add_argument( parser.add_argument(
'--data', '--data',
type=str, type=str,
......
...@@ -271,7 +271,7 @@ if __name__ == '__main__': ...@@ -271,7 +271,7 @@ if __name__ == '__main__':
parser.add_argument( parser.add_argument(
'--batch_size', type=int, default=256, help='batch size.') '--batch_size', type=int, default=256, help='batch size.')
parser.add_argument( parser.add_argument(
'--class_dim', type=int, default=1000, help='classify number.') '--class_dim', type=int, default=10, help='classify number.')
parser.add_argument( parser.add_argument(
'--data', '--data',
type=str, type=str,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册