[gradient checkpointing] lower tolerance for test (#652)

* lowe tolerance

* put model in eval mode

Author: patil-suraj

tests/test_models_unet.py

@@ -199,7 +199,7 @@ def prepare_init_args_and_inputs_for_common(self):
 
     def test_gradient_checkpointing(self):
         init_dict, inputs_dict = self.prepare_init_args_and_inputs_for_common()
-        model = self.model_class(**init_dict)
+        model = self.model_class(**init_dict).eval()
         model.to(torch_device)
 
         out = model(**inputs_dict).sample