Spaces:
Running
Running
fix start time initialization bug
Browse files- monitoring.py +6 -6
monitoring.py
CHANGED
|
@@ -40,17 +40,17 @@ class SmolLM3Monitor:
|
|
| 40 |
self.log_metrics = log_metrics
|
| 41 |
self.log_config = log_config
|
| 42 |
|
| 43 |
-
# Initialize
|
| 44 |
-
self.trackio_client = None
|
| 45 |
-
if self.enable_tracking:
|
| 46 |
-
self._setup_trackio(trackio_url, trackio_token)
|
| 47 |
-
|
| 48 |
-
# Experiment metadata
|
| 49 |
self.experiment_id = None
|
| 50 |
self.start_time = datetime.now()
|
| 51 |
self.metrics_history = []
|
| 52 |
self.artifacts = []
|
| 53 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
logger.info(f"Initialized monitoring for experiment: {experiment_name}")
|
| 55 |
|
| 56 |
def _setup_trackio(self, trackio_url: Optional[str], trackio_token: Optional[str]):
|
|
|
|
| 40 |
self.log_metrics = log_metrics
|
| 41 |
self.log_config = log_config
|
| 42 |
|
| 43 |
+
# Initialize experiment metadata first
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 44 |
self.experiment_id = None
|
| 45 |
self.start_time = datetime.now()
|
| 46 |
self.metrics_history = []
|
| 47 |
self.artifacts = []
|
| 48 |
|
| 49 |
+
# Initialize Trackio API client
|
| 50 |
+
self.trackio_client = None
|
| 51 |
+
if self.enable_tracking:
|
| 52 |
+
self._setup_trackio(trackio_url, trackio_token)
|
| 53 |
+
|
| 54 |
logger.info(f"Initialized monitoring for experiment: {experiment_name}")
|
| 55 |
|
| 56 |
def _setup_trackio(self, trackio_url: Optional[str], trackio_token: Optional[str]):
|