Keep NMS index gathering on cuda device #3070
build-wheels-aarch64-linux.yml
on: pull_request
generate-matrix
/
generate
4s
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.69 MB |
|