From 89daa9ff4dad18c3e5db00ce03c426630d266ccd Mon Sep 17 00:00:00 2001
From: Tim Chang <didwdidw0309@gmail.com>
Date: Tue, 6 Jul 2021 18:43:46 +0800
Subject: [PATCH] Fix error when training only 1 trial

---
 pybnn/lcnet.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/pybnn/lcnet.py b/pybnn/lcnet.py
index 031c4c2..7b69c39 100644
--- a/pybnn/lcnet.py
+++ b/pybnn/lcnet.py
@@ -87,5 +87,11 @@ def normalize_input(x, m=None, s=None):
 
         x_norm = deepcopy(x)
         x_norm[:, :-1] = (x[:, :-1] - m[:-1]) / s[:-1]
+        flag = True
+        for i in s[:-1]:
+            if i < 0.0001:
+                flag = False
+        if flag:
+            x_norm[:, :-1] = (x[:, :-1] - m[:-1]) / s[:-1]
 
         return x_norm, m, s