update
Browse files- app.py +1 -0
- requirements.txt +1 -1
app.py
CHANGED
|
@@ -10,6 +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_check.pt')['icnn'])
|
| 14 |
net_c = PretrainedConvNext("convnext_small_in22k").cpu()
|
| 15 |
net_c.load_state_dict(torch.load('./classifier_32.pt')['icnn'])
|
|
|
|
| 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=net_i.half()
|
| 14 |
net_i.load_state_dict(torch.load('./fp16_check.pt')['icnn'])
|
| 15 |
net_c = PretrainedConvNext("convnext_small_in22k").cpu()
|
| 16 |
net_c.load_state_dict(torch.load('./classifier_32.pt')['icnn'])
|
requirements.txt
CHANGED
|
@@ -6,7 +6,7 @@ fsspec
|
|
| 6 |
fvcore
|
| 7 |
huggingface-hub
|
| 8 |
matplotlib
|
| 9 |
-
numpy
|
| 10 |
opencv-python
|
| 11 |
omegaconf
|
| 12 |
pytorch-msssim
|
|
|
|
| 6 |
fvcore
|
| 7 |
huggingface-hub
|
| 8 |
matplotlib
|
| 9 |
+
numpy==1.26.4
|
| 10 |
opencv-python
|
| 11 |
omegaconf
|
| 12 |
pytorch-msssim
|