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

Adjust default batch size to 1.

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