-
Notifications
You must be signed in to change notification settings - Fork 1.2k
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
Torch auto classification task #1686
Conversation
Job PR-1686-0d5e7e0 is done. |
# mxnet and torch dispatcher | ||
dispatcher = None | ||
torch_model_list = None | ||
mxnet_model_list = list(get_model_list()) |
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.
requires mxnet here?
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 get_model_list() is part of gluoncv api and not require mxnet?
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.
no, model zoo still requires mxnet, only auto module can optionally use mxnet, otherwise gluoncv.torch is the right module
Job PR-1686-c163afc is done. |
@@ -20,10 +20,11 @@ | |||
from autogluon.core.scheduler.resource import get_cpu_count, get_gpu_count | |||
from autogluon.core.task.base import BaseTask | |||
from autogluon.core.searcher import RandomSearcher | |||
from gluoncv.model_zoo import get_model_list |
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.
here requires mxnet too
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.
btw, in package import should use relative import, from ... import xx
Job PR-1686-6e915f3 is done. |
Job PR-1686-5af5397 is done. |
Job PR-1686-c140d5e is done. |
Job PR-1686-cbed7ae is done. |
Thanks, I will merge it now |
No description provided.