You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Distillation code added for GLUE dataset in Hugging Face/transformers integration.
BERT quantization pipeline enabled for training and export to ONNX.
Changes:
Readme redesigned for better clarity on the repository's purpose.
All examples, notebooks, and scripts moved under the integrations directory.
Integrations for ultralytics/yolov3, ultralytics/yolov5, pytorch, keras, tensorflow reworked to match new integrations standards for better ease of use.
Resolved Issues:
rwightman/timm integration bugs dealing with checkpoint paths and loading models addressed.
tensorflow-gpu for tensorflow v1 now recognized correctly.
Neural Magic dependencies upgrade to intended bug versions instead of minor versions.
Known Issues:
Movement pruning is currently only working with FP16 training on GPUs, FP32 is diverging to NaN.