Closed evansakmar closed 3 years ago
I also have an incoming PR for this
Hi @ehsankf Thank you very much!
@beat-buesser
Hello,
This issue has not been opened by me.
Hi @ehsankf Thank you for notifying me, it must have been an autocomplete accident with user names starting with letter e. I wanted to mention @evansakmar above.
Describe the bug Whenever you run infer on MembershipInferenceBlackBox you get a CUDA runtime error.
The fit method seems to handle CUDA properly already. Noting the _tocuda method.
To Reproduce Steps to reproduce the behavior:
Expected behavior It should run properly without erroring out
Stacktrace RuntimeError: cuda runtime error (77) : an illegal memory access was encountered at /pytorch/aten/src/THC/THCCachingHostAllocator.cpp:278
System information (please complete the following information):