Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
@han-cai
han-cai authored Sep 26, 2023
1 parent f2a7b48 commit 69deca0
Showing 1 changed file with 50 additions and 52 deletions.
102 changes: 50 additions & 52 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,6 @@ random_subnet = ofa_network.get_active_subnet(preserve_weight=True)
ofa_network.set_active_subnet(ks=7, e=6, d=4)
manual_subnet = ofa_network.get_active_subnet(preserve_weight=True)
```
If the above scripts failed to download, you download it manually from [Google Drive](https://drive.google.com/drive/folders/10leLmIiMtaRu4J46KwrBaMydvQt0qFuI?usp=sharing) and put them under $HOME/.torch/ofa_nets/.

### Evaluate

Expand All @@ -82,7 +81,6 @@ Example: net, image_size = ofa_specialized('flops@[email protected]_finetune@75', p
from ofa.model_zoo import ofa_specialized
net, image_size = ofa_specialized(net_id, pretrained=True)
```
If the above scripts failed to download, you download it manually from [Google Drive](https://drive.google.com/drive/folders/1ez-t_DAHDet2fqe9TZUTJmvrU-AwofAt?usp=sharing) and put them under $HOME/.torch/ofa_specialized/.

### Evaluate

Expand All @@ -101,66 +99,66 @@ If the above scripts failed to download, you download it manually from [Google D
| ofa-resnet50D-09 | [email protected][email protected]_finetune@25 | 76.3 | 92.9 | 14.5M | 0.9B |
| ofa-resnet50D-06 | [email protected][email protected]_finetune@25 | 75.0 | 92.1 | 9.6M | 0.6B |
| **FLOPs** |
| ofa-595M| [flops@[email protected]_finetune@75](https://drive.google.com/file/d/1924Ta7J0xPKRef3Psa9X11C8BX872IOB/view?usp=sharing) | 80.0 | 94.9 | 9.1M | 595M |
| ofa-482M | [flops@[email protected]_finetune@75](https://drive.google.com/file/d/1IQLcolMQsctWbjcqk14CUMCV2HbIpuf7/view?usp=sharing) | 79.6 | 94.8 | 9.1M | 482M |
| ofa-389M | [flops@[email protected]_finetune@75](https://drive.google.com/file/d/1kmMj5ouaoij4atTcU8l6iOaAQW6gwsFX/view?usp=sharing) | 79.1 | 94.5 | 8.4M | 389M |
| ofa-595M| flops@[email protected]_finetune@75 | 80.0 | 94.9 | 9.1M | 595M |
| ofa-482M | flops@[email protected]_finetune@75 | 79.6 | 94.8 | 9.1M | 482M |
| ofa-389M | flops@[email protected]_finetune@75 | 79.1 | 94.5 | 8.4M | 389M |
| **LG G8** |
| ofa-lg-24 | [LG-G8_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1IK1TE-79aUbjfQz047XbU8nfXZvLGgQs/view?usp=sharing) | 76.4 | 93.0 | 5.8M | 230M |
| ofa-lg-16 | [LG-G8_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1D4zzNZDQY_KMX7dH1wwlbFWFVQ_ibEXp/view?usp=sharing) | 74.7 | 92.0 | 5.8M | 151M |
| ofa-lg-11 | [LG-G8_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1g6nUehJ48vBqX19GorqJLh-wg6wjH9Hp/view?usp=sharing) | 73.0 | 91.1 | 5.0M | 103M |
| ofa-lg-8 | [LG-G8_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1T6YAdRQsFzlDMOsZReZ7TFd-HlmJ3K0i/view?usp=sharing) | 71.1 | 89.7 | 4.1M | 74M |
| ofa-lg-24 | LG-G8_lat@[email protected]_finetune@25 | 76.4 | 93.0 | 5.8M | 230M |
| ofa-lg-16 | LG-G8_lat@[email protected]_finetune@25 | 74.7 | 92.0 | 5.8M | 151M |
| ofa-lg-11 | LG-G8_lat@[email protected]_finetune@25 | 73.0 | 91.1 | 5.0M | 103M |
| ofa-lg-8 | LG-G8_lat@[email protected]_finetune@25 | 71.1 | 89.7 | 4.1M | 74M |
| **Samsung S7 Edge** |
| ofa-s7edge-88 | [s7edge_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1u_N1zsJY1ZjjrDfOKz6l6IfEQHS0W0VM/view?usp=sharing) | 76.3 | 92.9 | 6.4M | 219M |
| ofa-s7edge-58| [s7edge_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1PsrxebWQ3fxyQjPU-NgIV5xcj-_T808H/view?usp=sharing) | 74.7 | 92.0 | 4.6M | 145M |
| ofa-s7edge-41 | [s7edge_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1ETdfEKlf_tdtW534npXplzcm85bp9KYF/view?usp=sharing) | 73.1 | 91.0 | 4.7M | 96M |
| ofa-s7edge-29 | [s7edge_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1EiIByUAG-KIgcBaqHVAcLZGr_TRbpspo/view?usp=sharing) | 70.5 | 89.5 | 3.8M | 66M |
| ofa-s7edge-88 | s7edge_lat@[email protected]_finetune@25 | 76.3 | 92.9 | 6.4M | 219M |
| ofa-s7edge-58| s7edge_lat@[email protected]_finetune@25 | 74.7 | 92.0 | 4.6M | 145M |
| ofa-s7edge-41 | s7edge_lat@[email protected]_finetune@25 | 73.1 | 91.0 | 4.7M | 96M |
| ofa-s7edge-29 | s7edge_lat@[email protected]_finetune@25 | 70.5 | 89.5 | 3.8M | 66M |
| **Samsung Note8** |
| ofa-note8-65 | [note8_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1UZ6NawJqyI2cWTuX6khM47wg4Gx4xCtV/view?usp=sharing) | 76.1 | 92.7 | 5.3M | 220M |
| ofa-note8-49 | [note8_lat@[email protected]_finetune@25](https://drive.google.com/file/d/11GyU7_AwLaGrNgGMoPIgavJ42u43jg8R/view?usp=sharing) | 74.9 | 92.1 | 6.0M | 164M |
| ofa-note8-31 | [note8_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1b-gZlGtZ88eETXfIE7OwjSLCao6AgGIx/view?usp=sharing) | 72.8 | 90.8 | 4.6M | 101M |
| ofa-note8-22 | [note8_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1mOwp6uAuAInIDqXOCru2ilEtmv7LOEj9/view?usp=sharing) | 70.4 | 89.3 | 4.3M | 67M |
| ofa-note8-65 | note8_lat@[email protected]_finetune@25 | 76.1 | 92.7 | 5.3M | 220M |
| ofa-note8-49 | note8_lat@[email protected]_finetune@25 | 74.9 | 92.1 | 6.0M | 164M |
| ofa-note8-31 | note8_lat@[email protected]_finetune@25 | 72.8 | 90.8 | 4.6M | 101M |
| ofa-note8-22 | note8_lat@[email protected]_finetune@25 | 70.4 | 89.3 | 4.3M | 67M |
| **Samsung Note10** |
| ofa-note10-64 | [note10_lat@[email protected]_finetune@75](https://drive.google.com/file/d/1Fniw7auRPYwn3LXkheb0jtVbz8P5zwzp/view?usp=sharing) | 80.2 | 95.1 | 9.1M | 743M |
| ofa-note10-50 | [note10_lat@[email protected]_finetune@75](https://drive.google.com/file/d/1xFVQaOV2-jdHzT8YI_lG3wMLjpWoS-nV/view?usp=sharing) | 79.7 | 94.9 | 9.1M | 554M |
| ofa-note10-41 | [note10_lat@[email protected]_finetune@75](https://drive.google.com/file/d/1Dg_MK2WE6iE4jYBlzN_2SOmaA2i1sGo5/view?usp=sharing) | 79.3 | 94.5 | 9.0M | 457M |
| ofa-note10-30 | [note10_lat@[email protected]_finetune@75](https://drive.google.com/file/d/1I31s1iYJsxh8eBPnpC9k4WyQwrvJvVN0/view?usp=sharing) | 78.4 | 94.2 | 7.5M | 339M |
| ofa-note10-22 | [note10_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1t2tf-y_OWZvBr-3nPx-PB7qasz-l5TrU/view?usp=sharing) | 76.6 | 93.1 | 5.9M | 237M |
| ofa-note10-16 | [note10_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1xgkW4he1midlvdCtWugL2VIg4TNLxa-l/view?usp=sharing) | 75.5 | 92.3 | 4.9M | 163M |
| ofa-note10-11 | [note10_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1mxvU0NSNFmfpxSuqcGhv5u_UbWh_ohqx/view?usp=sharing) | 73.6 | 91.2 | 4.3M | 110M |
| ofa-note10-08 | [note10_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1f8cmdR3-4hMXrEpf3oYecKrTJbTR4B3P/view?usp=sharing) | 71.4 | 89.8 | 3.8M | 79M |
| ofa-note10-64 | note10_lat@[email protected]_finetune@75 | 80.2 | 95.1 | 9.1M | 743M |
| ofa-note10-50 | note10_lat@[email protected]_finetune@75 | 79.7 | 94.9 | 9.1M | 554M |
| ofa-note10-41 | note10_lat@[email protected]_finetune@75 | 79.3 | 94.5 | 9.0M | 457M |
| ofa-note10-30 | note10_lat@[email protected]_finetune@75 | 78.4 | 94.2 | 7.5M | 339M |
| ofa-note10-22 | note10_lat@[email protected]_finetune@25 | 76.6 | 93.1 | 5.9M | 237M |
| ofa-note10-16 | note10_lat@[email protected]_finetune@25 | 75.5 | 92.3 | 4.9M | 163M |
| ofa-note10-11 | note10_lat@[email protected]_finetune@25 | 73.6 | 91.2 | 4.3M | 110M |
| ofa-note10-08 | note10_lat@[email protected]_finetune@25 | 71.4 | 89.8 | 3.8M | 79M |
| **Google Pixel1** |
| ofa-pixel1-143 | [pixel1_lat@[email protected]_finetune@75](https://drive.google.com/file/d/1I70UqBDJRbjwW5-q4bMAp-kUJwEAvnef/view?usp=sharing) | 80.1 | 95.0 | 9.2M | 642M |
| ofa-pixel1-132 | [pixel1_lat@[email protected]_finetune@75](https://drive.google.com/file/d/1mhtTGOJCkwF5n0bOluSdGeIDtCqoLd7l/view?usp=sharing) | 79.8 | 94.9 | 9.2M | 593M |
| ofa-pixel1-79 | [pixel1_lat@[email protected]_finetune@75](https://drive.google.com/file/d/1DHwA4JWPPT3XplT4yUrVXmtx-JBzjRG6/view?usp=sharing) | 78.7 | 94.2 | 8.2M | 356M |
| ofa-pixel1-58 | [pixel1_lat@[email protected]_finetune@75](https://drive.google.com/file/d/1yjO7JoXv8lj1_igPP1gkwfXSMXyKEnim/view?usp=sharing) | 76.9 | 93.3 | 5.8M | 230M |
| ofa-pixel1-40 | [pixel1_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1slT_GD7SeLTy2p7OPAHvc0NXpmTEkjNW/view?usp=sharing) | 74.9 | 92.1 | 6.0M | 162M |
| ofa-pixel1-28 | [pixel1_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1D5gwpF9UA31Bcz53DLQII1sCMLJZ_Ky4/view?usp=sharing) | 73.3 | 91.0 | 5.2M | 109M |
| ofa-pixel1-20 | [pixel1_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1sm5vBxjapw_wWrO8NFwiiCTZHCqZfVMx/view?usp=sharing) | 71.4 | 89.8 | 4.3M | 77M |
| ofa-pixel1-143 | pixel1_lat@[email protected]_finetune@75 | 80.1 | 95.0 | 9.2M | 642M |
| ofa-pixel1-132 | pixel1_lat@[email protected]_finetune@75 | 79.8 | 94.9 | 9.2M | 593M |
| ofa-pixel1-79 | pixel1_lat@[email protected]_finetune@75 | 78.7 | 94.2 | 8.2M | 356M |
| ofa-pixel1-58 | pixel1_lat@[email protected]_finetune@75 | 76.9 | 93.3 | 5.8M | 230M |
| ofa-pixel1-40 | pixel1_lat@[email protected]_finetune@25 | 74.9 | 92.1 | 6.0M | 162M |
| ofa-pixel1-28 | pixel1_lat@[email protected]_finetune@25 | 73.3 | 91.0 | 5.2M | 109M |
| ofa-pixel1-20 | pixel1_lat@[email protected]_finetune@25 | 71.4 | 89.8 | 4.3M | 77M |
| **Google Pixel2** |
| ofa-pixel2-62 | [pixel2_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1FexSgoOJ_tar1fXcyxbTr4ZNH564bYFh/view?usp=sharing) | 75.8 | 92.7 | 5.8M | 208M |
| ofa-pixel2-50 | [pixel2_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1TfDCZFGzlzQ65r3GhA0GIdVD49-hWKtZ/view?usp=sharing) | 74.7 | 91.9 | 4.7M | 166M |
| ofa-pixel2-35 | [pixel2_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1e4J0p4BlRI34lvo9K7JG5aCuWoLhGUbg/view?usp=sharing) | 73.4 | 91.1 | 5.1M | 113M |
| ofa-pixel2-25 | [pixel2_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1BWZY1UvvqPKIZmbHRgItWPcFwOs8Ke1_/view?usp=sharing) | 71.5 | 90.1 | 4.1M | 79M |
| ofa-pixel2-62 | pixel2_lat@[email protected]_finetune@25 | 75.8 | 92.7 | 5.8M | 208M |
| ofa-pixel2-50 | pixel2_lat@[email protected]_finetune@25 | 74.7 | 91.9 | 4.7M | 166M |
| ofa-pixel2-35 | pixel2_lat@[email protected]_finetune@25 | 73.4 | 91.1 | 5.1M | 113M |
| ofa-pixel2-25 | pixel2_lat@[email protected]_finetune@25 | 71.5 | 90.1 | 4.1M | 79M |
| **1080ti GPU (Batch Size 64)** |
| ofa-1080ti-27 | [1080ti_gpu64@[email protected]_finetune@25](https://drive.google.com/file/d/1_bxtAmBYVy-2joF2od_D1AXkHiGLav8y/view?usp=sharing) | 76.4 | 93.0 | 6.5M | 397M |
| ofa-1080ti-22 | [1080ti_gpu64@[email protected]_finetune@25](https://drive.google.com/file/d/16BvFoiJKwcXybp_LHK_2zvbykvVGmbL7/view?usp=sharing) | 75.3 | 92.4 | 5.2M | 313M |
| ofa-1080ti-15 | [1080ti_gpu64@[email protected]_finetune@25](https://drive.google.com/file/d/1mBgNJ8TEEr0gc4Zd5AXPx3swHgLxun9J/view?usp=sharing) | 73.8 | 91.3 | 6.0M | 226M |
| ofa-1080ti-12 | [1080ti_gpu64@[email protected]_finetune@25](https://drive.google.com/file/d/1QdiyXZ9HIxi_f3snpJNNQ5mvd8Wei2WN/view?usp=sharing) | 72.6 | 90.9 | 5.9M | 165M |
| ofa-1080ti-27 | 1080ti_gpu64@[email protected]_finetune@25 | 76.4 | 93.0 | 6.5M | 397M |
| ofa-1080ti-22 | 1080ti_gpu64@[email protected]_finetune@25 | 75.3 | 92.4 | 5.2M | 313M |
| ofa-1080ti-15 | 1080ti_gpu64@[email protected]_finetune@25 | 73.8 | 91.3 | 6.0M | 226M |
| ofa-1080ti-12 | 1080ti_gpu64@[email protected]_finetune@25 | 72.6 | 90.9 | 5.9M | 165M |
| **V100 GPU (Batch Size 64)** |
| ofa-v100-11 | [v100_gpu64@[email protected]_finetune@25](https://drive.google.com/file/d/1KOuj9Coyl8wTO-YzOtk3Ld-upo_UrIJD/view?usp=sharing) | 76.1 | 92.7 | 6.2M | 352M |
| ofa-v100-09 | [v100_gpu64@[email protected]_finetune@25](https://drive.google.com/file/d/1fUHJ0Ad-h0ReJpV8zvt4uoa4Kzp1Q2U1/view?usp=sharing) | 75.3 | 92.4 | 5.2M | 313M |
| ofa-v100-06 | [v100_gpu64@[email protected]_finetune@25](https://drive.google.com/file/d/1XhaH_8XUFzHwR1rNQ4bMaVC86DvMq6xj/view?usp=sharing) | 73.0 | 91.1 | 4.9M | 179M |
| ofa-v100-05 | [v100_gpu64@[email protected]_finetune@25](https://drive.google.com/file/d/1vTg7s5xd5vAcDHZu_eUNHkI8aFauIerU/view?usp=sharing) | 71.6 | 90.3 | 5.2M | 141M |
| ofa-v100-11 | v100_gpu64@[email protected]_finetune@25 | 76.1 | 92.7 | 6.2M | 352M |
| ofa-v100-09 | v100_gpu64@[email protected]_finetune@25 | 75.3 | 92.4 | 5.2M | 313M |
| ofa-v100-06 | v100_gpu64@[email protected]_finetune@25 | 73.0 | 91.1 | 4.9M | 179M |
| ofa-v100-05 | v100_gpu64@[email protected]_finetune@25 | 71.6 | 90.3 | 5.2M | 141M |
| **Jetson TX2 GPU (Batch Size 16)** |
| ofa-tx2-96 | [tx2_gpu16@[email protected]_finetune@25](https://drive.google.com/file/d/1oQoFNDZhTIreAInVGFKtiYX_CiAJmCZa/view?usp=sharing) | 75.8 | 92.7 | 6.2M | 349M |
| ofa-tx2-80 | [tx2_gpu16@[email protected]_finetune@25](https://drive.google.com/file/d/19lbh5U9s4l6KqxcQ2MnnQZZjMMdjz63c/view?usp=sharing) | 75.4 | 92.4 | 5.2M | 313M |
| ofa-tx2-47 | [tx2_gpu16@[email protected]_finetune@25](https://drive.google.com/file/d/1dqa14vskudMKF5MZtivdfgGDgztsLEFA/view?usp=sharing) | 72.9 | 91.1 | 4.9M | 179M |
| ofa-tx2-35 | [tx2_gpu16@[email protected]_finetune@25](https://drive.google.com/file/d/1HRnVmY0khszffEsv8ZtJlTaOcixzz9C4/view?usp=sharing) | 70.3 | 89.4 | 4.3M | 121M |
| ofa-tx2-96 | tx2_gpu16@[email protected]_finetune@25 | 75.8 | 92.7 | 6.2M | 349M |
| ofa-tx2-80 | tx2_gpu16@[email protected]_finetune@25 | 75.4 | 92.4 | 5.2M | 313M |
| ofa-tx2-47 | tx2_gpu16@[email protected]_finetune@25 | 72.9 | 91.1 | 4.9M | 179M |
| ofa-tx2-35 | tx2_gpu16@[email protected]_finetune@25 | 70.3 | 89.4 | 4.3M | 121M |
| **Intel Xeon CPU with MKL-DNN (Batch Size 1)** |
| ofa-cpu-17 | [cpu_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1eeGWBIj-3CPGlG_ZQGCmz7s1NinXM3Vu/view?usp=sharing) | 75.7 | 92.6 | 4.9M | 365M |
| ofa-cpu-15 | [cpu_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1YFL86KdUZbP9gJi9k_pzct3Fl_v-uG0Q/view?usp=sharing) | 74.6 | 92.0 | 4.9M | 301M |
| ofa-cpu-11 | [cpu_lat@[email protected]_finetune@25](https://drive.google.com/file/d/16cxbfpv9Z7V8-3ZBTjkufEVfIEb_GcDI/view?usp=sharing) | 72.0 | 90.4 | 4.4M | 160M |
| ofa-cpu-10 | [cpu_lat@[email protected]_finetune@25](https://drive.google.com/file/d/1_WFv_Ty105BBHsVDRJRL3idZJzHScv-4/view?usp=sharing) | 71.1 | 89.9 | 4.2M | 143M |
| ofa-cpu-17 | cpu_lat@[email protected]_finetune@25 | 75.7 | 92.6 | 4.9M | 365M |
| ofa-cpu-15 | cpu_lat@[email protected]_finetune@25 | 74.6 | 92.0 | 4.9M | 301M |
| ofa-cpu-11 | cpu_lat@[email protected]_finetune@25 | 72.0 | 90.4 | 4.4M | 160M |
| ofa-cpu-10 | cpu_lat@[email protected]_finetune@25 | 71.1 | 89.9 | 4.2M | 143M |

## How to train **OFA Networks**
```bash
Expand Down

0 comments on commit 69deca0

Please sign in to comment.