Update eval.py
Browse files
eval.py
CHANGED
|
@@ -59,5 +59,10 @@ def evaluate(config, epoch, pipeline):
|
|
| 59 |
|
| 60 |
|
| 61 |
|
|
|
|
|
|
|
| 62 |
|
| 63 |
-
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
|
| 60 |
|
| 61 |
|
| 62 |
+
def train_loop(config):
|
| 63 |
+
evaluate(config, 1, config.pipeline)
|
| 64 |
|
| 65 |
+
|
| 66 |
+
from accelerate import notebook_launcher
|
| 67 |
+
args = (config)
|
| 68 |
+
notebook_launcher(train_loop, args, num_processes=1)
|