Commit Graph

7 Commits (3d6bc42aa15b67883e3bf0f92df92fc7b74030b1)

Author SHA1 Message Date
Ross Wightman b9cfb64412 Support npz custom load for vision transformer hybrid models. Add posembed rescale for npz load.
4 years ago
Ross Wightman f31933cb37 Initial Vision Transformer impl w/ patch and hybrid variants. Refactor tuple helpers.
5 years ago
Ross Wightman 6c17d57a2c Fix some attributions, add copyrights to some file docstrings
5 years ago
Ross Wightman d0113f9cdb Fix a few issues that came up in tests
5 years ago
Ross Wightman 88129b2569 Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit.
5 years ago
Ross Wightman eb7653614f Monster commit, activation refactor, VoVNet, norm_act improvements, more
5 years ago
Ross Wightman 1a8f5900ab Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.
5 years ago