256 -> 512 -> 256

This commit is contained in:
Eric Wang 2023-03-28 08:34:36 -07:00
parent 804d22ad43
commit 3b79ea4029

View File

@ -33,7 +33,7 @@ def train(
micro_batch_size: int = 4,
num_epochs: int = 3,
learning_rate: float = 3e-4,
cutoff_len: int = 512,
cutoff_len: int = 256,
val_set_size: int = 2000,
# lora hyperparams
lora_r: int = 8,