-
Notifications
You must be signed in to change notification settings - Fork 2.7k
[Doc] Update add module doc #2067
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
Conversation
def train_step(self, | ||
data: Union[dict, tuple, list], | ||
optim_wrapper: OptimWrapper) -> Dict[str, torch.Tensor]: | ||
pass | ||
|
||
def val_step(self, | ||
data: Union[tuple, dict, list]) -> list: | ||
pass | ||
|
||
def test_step(self, | ||
data: Union[tuple, dict, list]) -> list: | ||
pass |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
BaseModel
in MMEngine implements xx_step, and mmseg didn't rewrite them.
I suggest that we should tell users what they should inherit from or rewrite, if they would like to add a segmentor
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
please rebase the laster dev-1.x
def aug_test(self, batch_inputs, batch_img_metas): | ||
"""Placeholder for augmentation test.""" | ||
pass |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
aug_test
isn't supported in openmmlab2.0, please remove it
type='MySegmentor' | ||
... | ||
) | ||
``` |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think it would be better to add how add a new data preprocessor in mmseg
7adfb8d
to
fd09554
Compare
* develop segmentor and remove custom optim * update segmentor example * add data preprocessor * refine intro
* develop segmentor and remove custom optim * update segmentor example * add data preprocessor * refine intro
Motivation
As title.
Modification