Improved Generalization of Weight Space Networks via Augmentations