Serge Durand comments

Results 3 comments of


                                            Serge Durand

[FEATURE] Allow parametrization of use of "device"

> How about adding a new function `set_device()` for change device? In this case, you can disable `.to(self.device)` by `set_device(None)`. Please refer to > > https://github.com/Harry24k/adversarial-attacks-pytorch/blob/8b35fb59b7447b9adf4ce8b7fa9fc8b2f9d637f4/torchattacks/attack.py#L57 > . Yes it...

[FEATURE] Allow parametrization of use of "device"

set_device works with multi-gpu on pytorch lightning with "ddp" strategy. Still have devices issue with dp, the model does not change device. But dp is discouraged anyway (https://pytorch-lightning.readthedocs.io/en/latest/accelerators/gpu_intermediate.html) so the...

JS error: Failed to load model class 'CaptumInsightsModel' from module 'jupyter-captum-insights'

I'm having the same problem. ```bash $ mamba list "captum|jupyter" # Name Version Build Channel captum 0.6.0 pyhd8ed1ab_0 conda-forge jupyter_client 8.1.0 pyhd8ed1ab_0 conda-forge jupyter_core 5.3.0 py310hff52083_0 conda-forge jupyter_events 0.6.3 pyhd8ed1ab_0...