Skip to content

Commit

Permalink
fix import error
Browse files Browse the repository at this point in the history
  • Loading branch information
NikolaiPetukhov committed Feb 28, 2024
1 parent 5fb8f7c commit 6b2323d
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions train/src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
from src.utils import verify_train_val_sets
from src.sly_to_yolov8 import check_bbox_exist_on_images, transform
from src.callbacks import on_train_batch_end
from src.dataset_cache import is_project_cached, download_project
from src.dataset_cache import download_project
from ultralytics import YOLO
import torch
from src.metrics_watcher import Watcher
Expand Down Expand Up @@ -565,7 +565,7 @@ def on_dataset_selected(new_dataset_ids):
elif new_dataset_ids != [] and reselect_data_button.is_hidden():
select_data_button.show()
update_globals(new_dataset_ids)
if is_project_cached(project_id):
if sly.is_cached(project_id):
use_cache_text.text = "Use cached data stored on the agent to optimize project downlaod"
else:
use_cache_text.text = "Cache data on the agent to optimize project download for future trainings"
Expand Down

0 comments on commit 6b2323d

Please sign in to comment.