Reinitializing weights vs units for maintaining plasticity in neural networks