lime-j commited on
Commit
1ffc7c5
·
1 Parent(s): 7cd868c
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('./fp16_checkpoint.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()
 
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