Skip to content

Commit

Permalink
Update ResNet tutorial (#317)
Browse files Browse the repository at this point in the history
Co-authored-by: ioangatop <[email protected]>
  • Loading branch information
roman807 and ioangatop authored Mar 19, 2024
1 parent 35bcf69 commit a217d7c
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions docs/user-guide/tutorials/evaluate_resnet.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,18 +21,18 @@ Now let's adapt the new `bach.yaml`-config to the new model:
path: timm.create_model
arguments:
model_name: resnet18
num_classes: &NUM_CLASSES 2
num_classes: &NUM_CLASSES 4
drop_rate: 0.0
pretrained: false
```
To reduce training time, lets overwrite some of the default parameters. In the terminal where you run ***eva***, set:
To reduce training time, lets overwrite some of the default parameters. In the terminal where you run *eva*, set:
```
export OUTPUT_ROOT=logs/resnet/bach
export MAX_STEPS=20
export LR_VALUE=0.1
export MAX_STEPS=50
export LR_VALUE=0.01
```
Now train and evaluate the model by running:
```
eva fit --config configs/vision/resnet18/bach.yaml
```
Once the run is complete, take a look at the results in `logs/resnet/bach/<session-id>/results.json`. How does the performance compare to the results observed in the previous tutorials?
Once the run is complete, take a look at the results in `logs/resnet/bach/<session-id>/results.json` and check out the tensorboard with `tensorboard --logdir logs/resnet/bach`. How does the performance compare to the results observed in the previous tutorials?

0 comments on commit a217d7c

Please sign in to comment.