Update train.py
Browse files
train.py
CHANGED
|
@@ -99,7 +99,7 @@ def main():
|
|
| 99 |
)
|
| 100 |
|
| 101 |
batch_size = 128
|
| 102 |
-
output_filename = 'meus_embeddings_e5_large.npy'
|
| 103 |
|
| 104 |
if os.path.exists(output_filename):
|
| 105 |
os.remove(output_filename)
|
|
|
|
| 99 |
)
|
| 100 |
|
| 101 |
batch_size = 128
|
| 102 |
+
output_filename = '/app/output/meus_embeddings_e5_large.npy'
|
| 103 |
|
| 104 |
if os.path.exists(output_filename):
|
| 105 |
os.remove(output_filename)
|