I’m using the default weights initialisation from PyTorch, which worked for the CNN (but not after I added GAP). Would it be a good idea to experiment with different weights initialisation methods? If so, which do you recommend? Thanks!