feat: adding resnet and formatting updates

This commit is contained in:
Lucas Oskorep
2023-04-06 00:37:46 -04:00
parent ce5939d8a9
commit dc427837f6
12 changed files with 87 additions and 37 deletions
+1 -2
View File
@@ -1,5 +1,4 @@
from collections import Callable
from typing import Callable
class ModelWrapper(object):
def __init__(self, model_func:Callable, model_preprocessor:Callable, name:str):