tamang0000
commited on
Commit
•
a4294e7
1
Parent(s):
bca8191
Update README.md
Browse files
README.md
CHANGED
@@ -26,7 +26,7 @@ max_seq_length = 512
|
|
26 |
per_device_train_batch_size = 2,
|
27 |
gradient_accumulation_steps = 4,
|
28 |
warmup_steps = 2,
|
29 |
-
max_steps =
|
30 |
learning_rate = 0.0005,
|
31 |
fp16 = not torch.cuda.is_bf16_supported(),
|
32 |
bf16 = torch.cuda.is_bf16_supported(),
|
|
|
26 |
per_device_train_batch_size = 2,
|
27 |
gradient_accumulation_steps = 4,
|
28 |
warmup_steps = 2,
|
29 |
+
max_steps = 20,
|
30 |
learning_rate = 0.0005,
|
31 |
fp16 = not torch.cuda.is_bf16_supported(),
|
32 |
bf16 = torch.cuda.is_bf16_supported(),
|