Skip to content

Commit

Permalink
hotfix link
Browse files Browse the repository at this point in the history
  • Loading branch information
mwalmsley committed Jul 22, 2023
1 parent a0d6b63 commit c3fc04b
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions docs/data_notes.rst
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,7 @@ Zoobot includes weights for the following pretrained models.
* - EfficientNetB0
- 224px
- 3
- Yes
- `Link <https://www.dropbox.com/s/19mooltlnofnj10crd88v/effnetb0_color_224px.ckpt?dl=0>`__
- `Link <https://www.dropbox.com/scl/fi/19mooltlnofnj10crd88v/effnetb0_color_224px.ckpt%26rlkey%3D2wczt74p8qmjvsf83sulrkdy4>`__
* - ResNet50
- 300px
- 1
Expand Down Expand Up @@ -108,7 +107,7 @@ We also include a few additional ad-hoc models `on Dropbox <https://www.dropbox.
Which model should I use?
--------------------------

We suggest the PyTorch EfficientNetB0 single-channel 224-pixel model for most users.
We suggest the PyTorch EfficientNetB0 single-channel 300-pixel model for most users.

Zoobot will prioritise PyTorch going forward. For more, see here.
The TensorFlow models currently perform just as well as the PyTorch equivalents but will not benefit from any future updates.
Expand All @@ -120,7 +119,7 @@ Color information does not improve overall performance at predicting GZ votes.
This is a little surprising, but we're confident it's true for our datasets (see the benchmarks folder for our tests).
However, it might be useful to include for other tasks where color is critical, such as hunting certain anomalous galaxies.

Larger input images (300px vs 224px) provide a very small boost in performance at predicting GZ votes, on our benchmarks.
Larger input images (300px vs 224px) provide a small boost in performance at predicting GZ votes.
However, the models require more memory and train/finetune slightly more slowly.
You may want to start with a 224px model and experiment with "upgrading" once you're happy everything works.

Expand Down

0 comments on commit c3fc04b

Please sign in to comment.