-
Notifications
You must be signed in to change notification settings - Fork 770
Open
Description
Is your feature request related to a problem? Please describe.
As the extract_levels implementation has been updated, the tutorial needs to be updated accordingly.
Describe the solution you'd like
Option 1: change extract_levels from [0, 1, 2, 3] to [3] to maintain the same model architecture as the pretrained weight.
Option 2: train a new model
Describe alternatives you've considered
Option 1 could be a quick fix during the training of option 2.
Additional context
N/A
Metadata
Metadata
Assignees
Labels
No labels