update
Browse files
app.py
CHANGED
|
@@ -10,7 +10,7 @@ channels = [64, 128, 256, 512]
|
|
| 10 |
layers = [2, 2, 4, 2]
|
| 11 |
num_subnet = 4
|
| 12 |
net_i = FullNet_NLP(channels, layers, num_subnet, 4,num_classes=1000, drop_path=0,save_memory=True, inter_supv=True, head_init_scale=None,kernel_size=3).cpu()
|
| 13 |
-
net_i.load_state_dict(torch.load('./
|
| 14 |
net_c = PretrainedConvNext("convnext_small_in22k").cpu()
|
| 15 |
net_c.load_state_dict(torch.load('./classifier_32.pt')['icnn'])
|
| 16 |
net_i.eval()
|
|
|
|
| 10 |
layers = [2, 2, 4, 2]
|
| 11 |
num_subnet = 4
|
| 12 |
net_i = FullNet_NLP(channels, layers, num_subnet, 4,num_classes=1000, drop_path=0,save_memory=True, inter_supv=True, head_init_scale=None,kernel_size=3).cpu()
|
| 13 |
+
net_i.load_state_dict(torch.load('./fp16_check.pt')['icnn'])
|
| 14 |
net_c = PretrainedConvNext("convnext_small_in22k").cpu()
|
| 15 |
net_c.load_state_dict(torch.load('./classifier_32.pt')['icnn'])
|
| 16 |
net_i.eval()
|
models/arch/__pycache__/RDnet_.cpython-310.pyc
CHANGED
|
Binary files a/models/arch/__pycache__/RDnet_.cpython-310.pyc and b/models/arch/__pycache__/RDnet_.cpython-310.pyc differ
|
|
|
models/arch/__pycache__/classifier.cpython-310.pyc
CHANGED
|
Binary files a/models/arch/__pycache__/classifier.cpython-310.pyc and b/models/arch/__pycache__/classifier.cpython-310.pyc differ
|
|
|
models/arch/__pycache__/focalnet.cpython-310.pyc
CHANGED
|
Binary files a/models/arch/__pycache__/focalnet.cpython-310.pyc and b/models/arch/__pycache__/focalnet.cpython-310.pyc differ
|
|
|
models/arch/__pycache__/modules_sig.cpython-310.pyc
CHANGED
|
Binary files a/models/arch/__pycache__/modules_sig.cpython-310.pyc and b/models/arch/__pycache__/modules_sig.cpython-310.pyc differ
|
|
|
models/arch/__pycache__/reverse_function.cpython-310.pyc
CHANGED
|
Binary files a/models/arch/__pycache__/reverse_function.cpython-310.pyc and b/models/arch/__pycache__/reverse_function.cpython-310.pyc differ
|
|
|