Spaces:
Sleeping
Sleeping
Commit
·
4633969
1
Parent(s):
1607add
Update trainer.py
Browse files- trainer.py +2 -2
trainer.py
CHANGED
|
@@ -9,7 +9,7 @@ import matplotlib.pyplot as plt
|
|
| 9 |
import cv2
|
| 10 |
import tensorflow as tf
|
| 11 |
from tqdm import tqdm
|
| 12 |
-
|
| 13 |
sn.set(font_scale=1.4)
|
| 14 |
|
| 15 |
class_names = ['buildings', 'forest', 'glacier', 'mountain', 'sea', 'street']
|
|
@@ -87,7 +87,7 @@ push_to_hub_keras(
|
|
| 87 |
log_dir="trainingsample",
|
| 88 |
commit_message="Optional commit message",
|
| 89 |
tags=["object-detection", "some_other_tag"],
|
| 90 |
-
include_optimizer=True
|
| 91 |
)
|
| 92 |
|
| 93 |
|
|
|
|
| 9 |
import cv2
|
| 10 |
import tensorflow as tf
|
| 11 |
from tqdm import tqdm
|
| 12 |
+
sac=os.getenv('accesstoken')
|
| 13 |
sn.set(font_scale=1.4)
|
| 14 |
|
| 15 |
class_names = ['buildings', 'forest', 'glacier', 'mountain', 'sea', 'street']
|
|
|
|
| 87 |
log_dir="trainingsample",
|
| 88 |
commit_message="Optional commit message",
|
| 89 |
tags=["object-detection", "some_other_tag"],
|
| 90 |
+
include_optimizer=True,token=sac
|
| 91 |
)
|
| 92 |
|
| 93 |
|