From 1f67fbdb80a52292e15dfd8a1c6537d17e97d400 Mon Sep 17 00:00:00 2001
From: William Falcon <falc@fb.com>
Date: Wed, 24 Jul 2019 15:23:38 -0400
Subject: [PATCH] removed old files

---
 tests/test_models.py | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/tests/test_models.py b/tests/test_models.py
index 5159b33e..1717adc5 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -35,6 +35,26 @@ def test_cpu_model():
 
     run_gpu_model_test(trainer_options, on_gpu=False)
 
+def test_all_features_cpu_model():
+    """
+    Make sure model trains on CPU
+    :return:
+    """
+
+    trainer_options = dict(
+        gradient_clip=1.0,
+        overfit_pct=0.20,
+        track_grad_norm=2,
+        print_nan_grads=True,
+        progress_bar=False,
+        experiment=get_exp(),
+        max_nb_epochs=1,
+        train_percent_check=0.4,
+        val_percent_check=0.4
+    )
+
+    run_gpu_model_test(trainer_options, on_gpu=False)
+
 
 def test_single_gpu_model():
     """