Merge pull request #633 from mv1388/amp-scaler-init-unittest

Amp scaler init unittest
mv1388 · Oct 31, 2020 · 00d98d9 · 00d98d9
2 parents f0afaa4 + 4c9468f
commit 00d98d9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_torchtrain/test_train_loop/test_train_loop.py b/tests/test_torchtrain/test_train_loop/test_train_loop.py
@@ -28,6 +28,7 @@ def test_init_values(self):
         self.assertIsInstance(train_loop.callbacks_handler, CallbacksHandler)
         self.assertEqual(train_loop.callbacks_handler.train_loop_obj, train_loop)
         self.assertFalse(train_loop.early_stop)
+        self.assertIsNone(train_loop.amp_scaler)
 
     def test_fit_mode_selection_single_gpu(self):
         gpu_mode, num_epochs, callbacks, grad_accumulation = self.execute_train_loop_fit_in_mode('single')