visualize warp field without self and self.forward. Make it dependent on only just grid, not warp object.

[vipulSharma18]

visualizing warp field causes computation on warping grid tensor in the training loop so can cause issues.
make it independent.