Upload folder using huggingface_hub
Browse files
src/dental_opg/components/model_trainer.py
CHANGED
|
@@ -62,7 +62,7 @@ class ModelTrainer:
|
|
| 62 |
"save_period": p.save_period,
|
| 63 |
"plots": p.plots,
|
| 64 |
"verbose": p.verbose,
|
| 65 |
-
"project": str(self.config.results_dir),
|
| 66 |
"name": "cavity_detection",
|
| 67 |
"exist_ok": True,
|
| 68 |
"device": p.device if p.device else "",
|
|
|
|
| 62 |
"save_period": p.save_period,
|
| 63 |
"plots": p.plots,
|
| 64 |
"verbose": p.verbose,
|
| 65 |
+
"project": str(Path(self.config.results_dir).resolve()),
|
| 66 |
"name": "cavity_detection",
|
| 67 |
"exist_ok": True,
|
| 68 |
"device": p.device if p.device else "",
|