Skip to content

feature(): Adding reset parameters - usefull for transfer learnings#31

Open
Rondineli wants to merge 1 commit intoOpenSportsLab:mainfrom
Rondineli:feature/transfer_learning_calf_model
Open

feature(): Adding reset parameters - usefull for transfer learnings#31
Rondineli wants to merge 1 commit intoOpenSportsLab:mainfrom
Rondineli:feature/transfer_learning_calf_model

Conversation

@Rondineli
Copy link

  • Working on some features for transfer learnings, I bumped to the scenario where those attributes were useful, so I added it as an option if it is useful for somebody else.

** Fixed couple of issues I ran into when working with a limited number of classes

# in case of a event not in EVENT_DICTIONARY move with the evaluation
# as label maybe None it wont evaluate for all classes
# but only the ones configured in the config model file
label = EVENT_DICTIONARY.get(event)
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This was failing if you set only limited classes to the dataset it tries to evaluate everything but the dictionary will contain only the limited list of classes.

cont = False

# Initializing label var in case the cont is False dont thrown an error
label = None
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

same here, it would fail as cont = False the label will not be declared, causing the code to crash.


# Adding reset_backbone, reset_neck and freeze_backbone will allow
# in case of transfer learning domains
if getattr(cfg.model, "freeze_backbone", False):
Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

added freeze_backbone, freeze_neck and reset_head usefull features for transfer learning scenarios, not sure this is ok from the maintainers to add this feature but thought to do it just in case.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant