From 33f9a0c078975b5eed860a8fbac5ac5683e542e7 Mon Sep 17 00:00:00 2001
From: Jeffrey Ling <jeffrey.r.ling@gmail.com>
Date: Mon, 20 Mar 2017 20:39:53 -0400
Subject: [PATCH] .

---
 main.lua | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/main.lua b/main.lua
index b1230a2..17e74b6 100644
--- a/main.lua
+++ b/main.lua
@@ -253,6 +253,7 @@ function load_data()
     opt.has_dev = 1
     dev = f:read('dev'):all()
     dev_label = f:read('dev_label'):all()
+    assert(torch.max(dev_label) <= opt.num_classes, 'more valid classes than train')
   end
   if f:read('test'):dataspaceSize()[1] == 0 then
     opt.has_test = 0
@@ -260,6 +261,7 @@ function load_data()
     opt.has_test = 1
     test = f:read('test'):all()
     test_label = f:read('test_label'):all()
+    assert(torch.max(test_label) <= opt.num_classes, 'more test classes than train')
   end
   print('data loaded!')