Spaces:
Sleeping
Sleeping
Update trainDistilBERT.py
Browse files- trainDistilBERT.py +2 -0
trainDistilBERT.py
CHANGED
@@ -16,6 +16,8 @@ import os
|
|
16 |
import random
|
17 |
import numpy as np
|
18 |
|
|
|
|
|
19 |
config = {
|
20 |
"mdl": "distilbert-base-uncased",
|
21 |
"epochs": 5,
|
|
|
16 |
import random
|
17 |
import numpy as np
|
18 |
|
19 |
+
WANDB_MODE=offline
|
20 |
+
|
21 |
config = {
|
22 |
"mdl": "distilbert-base-uncased",
|
23 |
"epochs": 5,
|