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.

Sign up for GitHub

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

Jump to bottom

ENH: update app dockerfiles + add mypy/flake8 to github ci + support weights from timm #51

Merged
merged 18 commits into from
Dec 26, 2022
Merged

ENH: update app dockerfiles + add mypy/flake8 to github ci + support weights from timm #51

merged 18 commits into from
Dec 26, 2022

Conversation

kaczmarj
Copy link
Member

This pull request does the following:

  • use the docker image tag 'kaczmarj/wsinfer'
  • in dockerfiles, copy wsinfer to /opt/wsinfer
  • add shields for python versions and pypi version to readme
  • add version key to yaml files. right now, we only support "1.0"
  • support loading weights made by timm. the state dict file has a key "state_dict" with the weights. Use that if it exists.
  • add flake8 and mypy checks to github ci.
  • fix all flake8 and mypy errors.
  • move patching cli to the 'wsinfer.cli' namespace.

@kaczmarj
use image tag kaczmarj/wsinfer
15a594d
@kaczmarj
use workdir /opt/wsinfer and use default cmd --help
409f27d
@kaczmarj
add shields for python versions and pypi version
18bc190
@kaczmarj
add version key to yaml configs
dae4efc
@kaczmarj
load weights from 'state_dict' key if it exists
786f329 
When using timm for model training, weights are saved to the state_dict
key of the state dict file. This commit uses the parameters in the
state_dict key if it exists.
@kaczmarj
move patching cli to wsinfer.cli namespace
9bee9da
@kaczmarj
rm unused imports + split long lines
c2deb90
@kaczmarj
ignore types for missing imports
ea5204c
@kaczmarj
split long comment
5742c84
@kaczmarj
rm pretrained_settings dict
cff53c8
@kaczmarj
add type annots to __getitem__
1084ce8
@kaczmarj
move patching cli to wsinfer.cli namespace
93e827b
@kaczmarj
fix forward pass types for mypy
bc5d663 
It seems that if I use the same variable name for the tensor or pil
image throughout the function definition, the Union type does not get
restricted to a Tensor type. But if I change the name of the variable,
mypy passes.
@kaczmarj
run flake8 and mypy
a96d5f6
@kaczmarj
use Weights().load_model() to download weights
ce82bea
@kaczmarj
add instructions to list available models
869b879
@kaczmarj
add types- packages to dev extras
d193c89
@kaczmarj
unindent a line
858aba7
@kaczmarj kaczmarj merged commit bc41cab into main Dec 26, 2022
@kaczmarj kaczmarj deleted the enh/docker-wsinfer branch December 26, 2022 23:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@kaczmarj