Nota-NetsPresso/netspresso-trainer
A library for training, compressing and deploying computer vision models (including ViT) with edge devices
PythonApache-2.0
Issues
- 1
[Sprint] Add MIT re-writed version of YOLOv7, 9
#582 opened by hglee98 - 0
[Sprint] Add feature to log predictions
#599 opened by hglee98 - 0
[Sprint] Refactor RT-DETR blocks & enable fused and reparameterzed block when deploying the model
#529 opened by hglee98 - 0
[Sprint] Add features to log precision and recall metric for each class. (detection task)
#578 opened by hglee98 - 0
- 0
- 0
- 0
[Sprint] Evaluation with tensorflow lite
#560 opened by hglee98 - 0
[Sprint] Refactoring and enhance Metric modules
#561 opened by illian01 - 0
[Feature Request] Add MultiStepLR scheduler
#558 opened by hglee98 - 0
- 0
- 0
- 0
[BUG] fx graphmodule does not have deploy method
#542 opened by illian01 - 0
[Sprint] Add onnx export tool
#540 opened by illian01 - 0
[Sprint] Evaluation with onnxruntime
#536 opened by illian01 - 0
[Sprint] Remove fx_model_path field in model config
#537 opened by illian01 - 1
- 0
[Sprint] Enable using different learning rate per modules and add option for no weight decay on bias and norm weight
#528 opened by illian01 - 1
[Sprint] Enable convert MobileViT to TensorRT
#473 opened by illian01 - 1
[Sprint] Enable convert SegFormer to TensorRT
#469 opened by illian01 - 1
[Feature Request] Add support for RT-DETR
#480 opened by hglee98 - 0
[Sprint] Add backbone: MobileNetV4
#508 opened by illian01 - 0
[Sprint] Add option for opset of onnx export
#478 opened by illian01 - 1
[Sprint] Enable to set onnx export opset version
#524 opened by illian01 - 3
[BUG] Failed segmentation resnet-50 training
#511 opened by Only-bottle - 0
- 0
[BUG] GradScaler scales the mean loss values
#512 opened by hglee98 - 0
[BUG] Add scipy library in requirements.txt
#515 opened by Only-bottle - 0
[Feature Request] Gradient Clipping
#505 opened by hglee98 - 0
- 0
[Sprint] Enhance ResNet to compatible with RT-DETR
#496 opened by illian01 - 0
[Feature Request] Add Objects365 open dataset
#481 opened by hglee98 - 1
[Sprint] Add error_stat on training summary
#486 opened by illian01 - 0
[Sprint] Update postprocessor page of document
#484 opened by illian01 - 0
[Sprint] Documentation update
#479 opened by illian01 - 0
[Sprint] Enable convert PIDNet to TensorRT
#476 opened by illian01 - 0
[Sprint] Enable convert ViT to TensorRT
#474 opened by illian01 - 0
[Sprint] Enable convert EfficientFormer to TensorRT
#463 opened by illian01 - 0
[Feature Request] Add support for YOLOX-light models
#466 opened by hglee98 - 1
- 0
[Sprint] Refactoring segmentation mask loading logic
#459 opened by illian01 - 0
[Sprint] Refactoring ImageSaver
#457 opened by illian01 - 0
[BUG] Wrong segmentation metric logic
#454 opened by illian01 - 0
[Sprint] Add copyright comments for entire project
#450 opened by illian01 - 0
[Sprint] Update docs Benchmarks & Checkpoints page
#442 opened by illian01 - 0
[Sprint] Remove thop dependency
#443 opened by illian01 - 0
- 0
- 0
[BUG] Wrong dataset path check
#427 opened by illian01