-
Notifications
You must be signed in to change notification settings - Fork 47
Open
Description
When training complete there is a no folder exist best_cars_encoders
https://qdrant.tech/articles/cars-recognition/
But in evulator.py os.path.join(os.path.dirname(file), "best_cars_encoders") want to load best_cars_encoders
def eval_tuned_encoder(dataset, device):
print("Evaluating tuned encoder...")
tuned_cars_model = SimilarityModel.load(
os.path.join(os.path.dirname(__file__), "best_cars_encoders")
).to(device)
tuned_cars_model.eval()
result = Quaterion.evaluate(
evaluator=Evaluator(
metrics=RetrievalRPrecision(),
sampler=GroupSampler(sample_size=1000, device=device, log_progress=True),
),
model=tuned_cars_model,
dataset=dataset,
)
print(result)
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels