Keep NMS index gathering on cuda device #3067
build-wheels-aarch64-linux.yml
on: pull_request
generate-matrix
/
generate
7s
Matrix: build
Artifacts
Produced during runtime
Name | Size | |
---|---|---|
pytorch_vision__3.9_cpu_aarch64
|
2.37 MB |
|
pytorch_vision__3.9_cu126_aarch64
|
7.45 MB |
|
pytorch_vision__3.9_cu128_aarch64
|
8.7 MB |
|