Hi! This callback was totally new for me but I loved the idea! So as usually I went to the documentation and tried implementing it with a very minimalist example (resnet18 for mnist) but I may be missing something. I have seen there are 2 threads for this topic on our forum. Can somebody take a quick look at my question here? Thanks!