Skip to content

Commit

Permalink
update tutorial
Browse files Browse the repository at this point in the history
  • Loading branch information
moreib committed Aug 30, 2024
1 parent 5bc565a commit 34fe507
Show file tree
Hide file tree
Showing 2 changed files with 6,150 additions and 1,221 deletions.
4 changes: 2 additions & 2 deletions tutorials/automlschool24/config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ defaults:
- base_config
- override dataset: mnist # Dataset configuration name
- override features: identity # Feature extractor configuration name (use identity for vision datasets)
- override model: timm_mobilenetv3_small_075 # Neural network name (for now timm_resnet50 or timm_efficientnet_lite1)
# - override model: timm_mobilenetv3_small_075 # Neural network name (for now timm_resnet50 or timm_efficientnet_lite1)
- override scheduler: 1cycle # learning rate scheduler config name
- override optimizer: adamw # Optimizer config name
- override normalizer: null # Feature normalizer (used for quantized neural networks)
Expand All @@ -41,4 +41,4 @@ trainer:
scheduler:
max_lr: 0.001

fx_mac_summary: False
fx_mac_summary: True
Loading

0 comments on commit 34fe507

Please sign in to comment.