Commit 80b80fad authored by Wes Brown's avatar Wes Brown

Adjust default batch size to 1.

parent fe34cc23
......@@ -43,7 +43,7 @@ spec:
# Model training parameters
# Batch size.
- name: batch_size
value: 4
value: 1
- name: gas
value: 1
- name: random_seed
......
......@@ -266,7 +266,7 @@ parser.add_argument('--lr', type=float, help='learning rate', default=2e-4)
parser.add_argument('--end_lr', type=float, help='end learning rate',
default=2e-4)
parser.add_argument('--warmup', type=int, help='warmup steps', default=10)
parser.add_argument('--bs', type=int, help='batch size', default=4)
parser.add_argument('--bs', type=int, help='batch size', default=1)
parser.add_argument('--gas', type=int, help='gas', default=1)
parser.add_argument('--seed', type=int, help="Random seed value",
default=42)
......
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