Commit 7598a587 authored by zhiyang.zhou's avatar zhiyang.zhou

minor change

parent ca355dee
Pipeline #192 canceled with stages
...@@ -27,6 +27,9 @@ parser.add_argument('--schedule', type=int, nargs='+', default=[100, 150], ...@@ -27,6 +27,9 @@ parser.add_argument('--schedule', type=int, nargs='+', default=[100, 150],
parser.add_argument('--gamma', type=float, default=0.1, help='LR is multiplied by gamma on schedule.') parser.add_argument('--gamma', type=float, default=0.1, help='LR is multiplied by gamma on schedule.')
parser.add_argument('--model_dir', default='dnn_models/tongue_modes/', help='directory of model for saving checkpoint') parser.add_argument('--model_dir', default='dnn_models/tongue_modes/', help='directory of model for saving checkpoint')
parser.add_argument('--save_freq', '-s', default=1, type=int, metavar='N', help='save frequency') parser.add_argument('--save_freq', '-s', default=1, type=int, metavar='N', help='save frequency')
parser.add_argument('--log_interval', type=int, default=100, metavar='N',
help='how many batches to wait before logging training status')
args = parser.parse_args() args = parser.parse_args()
NUM_CLASSES = 7 NUM_CLASSES = 7
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment