Benjamin Antin comments

Results 7 comments of


                                            Benjamin Antin

No kernel image is available for execution on the device

I'm having the same issue on an HPC using a singularity container. I'm not the admin so I can't update the nvidia driver. If a jax workaround that still allows...

test script does not properly normalize denoised movie

Hi, thanks for the response! 1. I trained the model on my dataset. 2. The scale mismatch seems to occur across all frames 3. Training command: `python -m src.train --exp_name...

test script does not properly normalize denoised movie

Hi, you're talking about changing the patch size used in test.py right? I changed that to the values you suggested but it didn't fix the problem.

test script does not properly normalize denoised movie

Thanks, will be happy to send a section of the video later today. However, I'm trying to do a bit of debugging myself. In the test script, each patch passed...

test script does not properly normalize denoised movie

Hi @SteveJayH. I tried the same pipeline on synthetic data and it seemed to work. I probably had a bug in my code -- I will try rerunning my analysis...

test script does not properly normalize denoised movie

Thanks! That is helpful. And another clarification -- I've trained a model with patch size [32 32 32] and patch interval [1 32 32]. I'd like to reduce the patch...

Tests failing due to incompatibility between TFP and JAX 0.7

Also had this problem. Was able to temporarily get around it by installing tfp-nightly (as detailed in https://github.com/tensorflow/probability/issues/2009)