At the moment I am trying the functionality of the framework on the Kirsch problem. And for a good solution, it requires a lot of density around the cutout. Is it possible to adaptively compact the sample closer to the edges during training? At the moment I have divided the area into two parts - the small area around the cutout and everything else. Are there any smarter ways?