Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • YosuaMichael's avatar
    Add SWAG model weight that only the linear head is finetuned to ImageNet1K (#5793) · 3fa24148
    YosuaMichael authored 
    * Add SWAG model that only the linear classifier head is finetuned with frozen trunk weight

* Add accuracy from experiments

* Change name from SWAG_LC to SWAG_LINEAR

* Add comment on SWAG_LINEAR weight

* Remove the comment docs (moved to PR description), and add the PR url as recipe. Also change name of previous swag model to SWAG_E2E_V1
    3fa24148
regnet.py 40.1 KB