Save Image Search Model to Disk

This commit is contained in:
Debanjum Singh Solanky
2022-01-14 17:09:18 -05:00
parent 934ec233b0
commit 510faa1904
4 changed files with 15 additions and 7 deletions

View File

@@ -49,6 +49,7 @@ class AsymmetricConfig(ConfigBase):
class ImageSearchTypeConfig(ConfigBase):
encoder: Optional[str]
model_directory: Optional[Path]
class SearchTypeConfig(ConfigBase):
asymmetric: Optional[AsymmetricConfig]