Browse Source

fixed some import for convenience

Dimitri Korsch 6 years ago
parent
commit
5ebc6a2967
1 changed files with 9 additions and 2 deletions
  1. 9 2
      cvfinetune/parser/__init__.py

+ 9 - 2
cvfinetune/parser/__init__.py

@@ -1,8 +1,15 @@
-from cvfinetune.parser.base import FineTuneParser, default_factory
+from cvfinetune.parser.base import FineTuneParser
+from cvfinetune.parser.base import add_dataset_args
+from cvfinetune.parser.base import add_model_args
+from cvfinetune.parser.base import add_training_args
+from cvfinetune.parser.base import default_factory
 
 
 __all__ = [
+	"add_dataset_args",
+	"add_model_args",
+	"add_training_args",
+	"default_factory",
 	"FineTuneParser",
-	"default_factory"
 ]