Batch size changes, mostly

2 files changed, 6 insertions, 6 deletions

diff --git a/config/lstm-frigo-irc.py b/config/lstm-frigo-irc.py
index 2d0bf3a..4ae21e2 100644
--- a/config/lstm-frigo-irc.py
+++ b/config/lstm-frigo-irc.py
@@ -3,14 +3,14 @@ from blocks.bricks import Tanh
 
 from model.lstm import Model
 
-dataset = 'data/logcompil-2016-03-07.txt'
+dataset = 'data/logcompil.txt'
 io_dim = 256
 
 # An epoch will be composed of 'num_seqs' sequences of len 'seq_len'
 # divided in chunks of lengh 'seq_div_size'
-num_seqs = 100
-seq_len = 2000
-seq_div_size = 100
+num_seqs = 50
+seq_len = 5000
+seq_div_size = 200
 
 hidden_dims = [1024, 1024, 1024]
 activation_function = Tanh()
@@ -26,7 +26,7 @@ l1_reg = 0
 step_rule = AdaDelta()
 
 # parameter saving freq (number of batches)
-monitor_freq = 10
+monitor_freq = 100
 save_freq = 100
 
 # used for sample generation and IRC mode
diff --git a/ircbot.py b/ircbot.py
index ac329eb..7477691 100755
--- a/ircbot.py
+++ b/ircbot.py
@@ -107,7 +107,7 @@ class IRCClient(SimpleIRCClient):
             else:
                 # feed phrase to bot
                 prob, = pred_f(self.str2data(s0+'\n'))
-                if self.nick in msg:
+                if any(x in msg.lower() for x in [self.nick, 'frigal']):
                     self.pred_until(pred_f, prob, '\t') 
                     prob, = pred_f(self.str2data(nick+': '))
                     rep = nick + ': ' + self.pred_until(pred_f, prob)