MadryLab / trak

A fast, effective data attribution method for neural networks in PyTorch
https://trak.csail.mit.edu/
MIT License
169 stars 22 forks source link

CudaProjector gives `CUDA error: too many resources requested for launch` on V100 #26

Closed kristian-georgiev closed 1 year ago

kristian-georgiev commented 1 year ago

Resolved by https://github.com/MadryLab/trak/commit/0b144dfd5ab97c4fbe0571341bc44f5dac8a435d.