torch-submod icon indicating copy to clipboard operation
torch-submod copied to clipboard

Published 20 hours ago • verified publisher icon josipd
→

Metadata

A PyTorch library for differentiable submodular minimization.

Results 2 torch-submod issues
Sort by recently updated
recently updated
newest added

GPU version of connected componet

1 comment

Is there API for GPU version connected componet

mathfinder avatar mathfinder

GPU version

Thanks a lot for sharing this code. Is there any work done or being done to convert the CPU code to GPU implementation? Because I am considering incorporating this layer...

ccj5351 avatar ccj5351

← Metadata

54
Stars
10
Forks
Watchers

Owner

verified publisher icon josipd

Metadata

A PyTorch library for differentiable submodular minimization.

Back