Update README.md
Browse files
README.md
CHANGED
|
@@ -18,8 +18,6 @@ MobiusNet introduces a fundamentally different approach to neural network design
|
|
| 18 |
|
| 19 |
| Model | Params | GFLOPs | Tiny ImageNet |
|
| 20 |
|-------|--------|--------|---------------|
|
| 21 |
-
| ResNet-18 | 11M | 1.8 | 50-55% |
|
| 22 |
-
| MobiusNet-M | 14.6M | 2.69 | 55.4% |
|
| 23 |
| MobiusNet-Base | 33.7M | 2.69 | TBD |
|
| 24 |
|
| 25 |
## Installation
|
|
@@ -220,22 +218,9 @@ wget http://cs231n.stanford.edu/tiny-imagenet-200.zip
|
|
| 220 |
unzip tiny-imagenet-200.zip -d ./data/
|
| 221 |
```
|
| 222 |
|
| 223 |
-
## HuggingFace Integration
|
| 224 |
-
|
| 225 |
-
Checkpoints auto-upload to HuggingFace Hub:
|
| 226 |
-
|
| 227 |
-
```python
|
| 228 |
-
# Set token in Colab
|
| 229 |
-
from google.colab import userdata
|
| 230 |
-
token = userdata.get('HF_TOKEN')
|
| 231 |
-
|
| 232 |
-
# Or environment variable
|
| 233 |
-
export HF_TOKEN=your_token_here
|
| 234 |
-
```
|
| 235 |
-
|
| 236 |
## License
|
| 237 |
|
| 238 |
-
|
| 239 |
|
| 240 |
## Citation
|
| 241 |
|
|
|
|
| 18 |
|
| 19 |
| Model | Params | GFLOPs | Tiny ImageNet |
|
| 20 |
|-------|--------|--------|---------------|
|
|
|
|
|
|
|
| 21 |
| MobiusNet-Base | 33.7M | 2.69 | TBD |
|
| 22 |
|
| 23 |
## Installation
|
|
|
|
| 218 |
unzip tiny-imagenet-200.zip -d ./data/
|
| 219 |
```
|
| 220 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 221 |
## License
|
| 222 |
|
| 223 |
+
Apache 2.0
|
| 224 |
|
| 225 |
## Citation
|
| 226 |
|