Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ import json
|
|
4 |
import torch
|
5 |
|
6 |
# Define hyperparameters
|
7 |
-
learning_rate =
|
8 |
batch_size = 8 # Smaller batch size to allow for more precise updates
|
9 |
epochs = 4 # Slightly more training epochs
|
10 |
max_seq_length = 256 # Smaller sequence length, especially if the majority of your questions and contexts are shorter
|
|
|
4 |
import torch
|
5 |
|
6 |
# Define hyperparameters
|
7 |
+
learning_rate = 5e-5 # Slightly lower learning rate
|
8 |
batch_size = 8 # Smaller batch size to allow for more precise updates
|
9 |
epochs = 4 # Slightly more training epochs
|
10 |
max_seq_length = 256 # Smaller sequence length, especially if the majority of your questions and contexts are shorter
|