Update README.md
Browse files
README.md
CHANGED
|
@@ -6,10 +6,11 @@ reproducing: "Ensemble everything everywhere: Multi-scale aggregation for advers
|
|
| 6 |
|
| 7 |
source code and usage examples: https://github.com/ETH-DISCO/self-ensembling
|
| 8 |
|
|
|
|
|
|
|
| 9 |
hyperparameters:
|
| 10 |
|
| 11 |
```
|
| 12 |
-
- architecture = based on Torchvision's Resnet152 default implementation
|
| 13 |
- criterion = torch.nn.CrossEntropyLoss()
|
| 14 |
- optimizer = torch.optim.AdamW
|
| 15 |
- scaler = GradScaler
|
|
|
|
| 6 |
|
| 7 |
source code and usage examples: https://github.com/ETH-DISCO/self-ensembling
|
| 8 |
|
| 9 |
+
architecture based on Torchvision's Resnet152 default implementation
|
| 10 |
+
|
| 11 |
hyperparameters:
|
| 12 |
|
| 13 |
```
|
|
|
|
| 14 |
- criterion = torch.nn.CrossEntropyLoss()
|
| 15 |
- optimizer = torch.optim.AdamW
|
| 16 |
- scaler = GradScaler
|