tweaks and add a simple test

2023-07-23 14:52:08 +00:00
parent f499d9d2b5
commit 9414e7a45e
7 changed files with 73 additions and 48 deletions
@@ -44,8 +44,6 @@ for k,v in list(state_dict.items()):
        state_dict[k[len(unwanted_prefix):]] = state_dict.pop(k)
 model.load_state_dict(state_dict, strict=False)

-model.export() # model.bin
-
 model.eval()
 model.to(device)
 if compile: