Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

model_choice.py: remove define_model() function #461

Closed
wants to merge 4 commits into from

Conversation

remtav
Copy link
Collaborator

@remtav remtav commented Jan 23, 2023

Fixes #460

rtavon added 4 commits January 19, 2023 16:59
… if url

default_multiclass.yaml: add checkpoint_dir param
default_binary.yaml: idem
model_choice.py:
- remove load state dict path from define_model
- read_checkpoint: set default "update=False", not True
train_segmentation.py: adjust usage of define_model
test_models.py: idem
inference_segmentation.py: refer directly to define_model_architecture()
train_segmentation.py: idem
test_models.py: idem
default_multiclass.yaml: idem
@remtav
Copy link
Collaborator Author

remtav commented Jan 31, 2023

@valhassan mentionned that define_model() seems still useful in train_segmentation.py to handle model architecture definition and weight loading into a single function. Define_model() will live, for now :)

@remtav remtav closed this Jan 31, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
3 participants