You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Benjamin Bossan a5b01ec04e Add type annotations to _registry.py ... Description Add type annotations to _registry.py so that they will pass mypy --strict. Comment I was reading the code and felt that this module would be easier to understand with type annotations. Therefore, I went ahead and added the annotations. The idea with this PR is to start small to see if we can align on _how_ to annotate types. I've seen people in the past disagree on how strictly to annotate the code base, so before spending too much time on this, I wanted to check if you agree, Ross. Most of the added types should be straightforward. Some notes on the non-trivial changes: - I made no assumption about the fn passed to register_model, but maybe the type could be stricter. Are all models nn.Modules? - If I'm not mistaken, the type hint for get_arch_name was incorrect - I had to add a # type: ignore to model.__all__ = ... - I made some minor code changes to list_models to facilitate the typing. I think the changes should not affect the logic of the function. - I removed list from list(sorted(...)) because sorted returns always a list.		2 years ago
..
data	Improve support for custom dataset label name/description through HF hub export, via pretrained_cfg	2 years ago
layers	Torchscript fixes/hacks for rms_norm, refactor ParallelScalingBlock with manual combination of input projections, closer paper match	2 years ago
loss	…
models	Add type annotations to _registry.py	2 years ago
optim	Add multi-tensor (foreach) version of Lion in style of upcoming PyTorch 2.0 optimizers	2 years ago
scheduler	…
utils	…
__init__.py	…
version.py	Version bump	2 years ago