creating model builder for faster model tweaking and iteration. Increased threads for training to better feed the GPU images from the image generator.

This commit is contained in:
Lucas
2022-05-25 19:47:30 -04:00
parent 5990351a47
commit 616c99c7f5
12 changed files with 189 additions and 171 deletions
+7
View File
@@ -0,0 +1,7 @@
from collections import Callable
class ModelWrapper(object):
def __init__(self, model_func:Callable, model_preprocessor:Callable):
self.model_func = model_func
self.model_preprocessor = model_preprocessor