From fd5cafe47faee22f012e57ab0aec2d4b3e612376 Mon Sep 17 00:00:00 2001
From: Matt Harvey <harveym@gmail.com>
Date: Mon, 26 Feb 2018 15:00:28 -0800
Subject: [PATCH] Minor
---
validate_rnn.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/validate_rnn.py b/validate_rnn.py
index adf3dd6..8368893 100644
--- a/validate_rnn.py
+++ b/validate_rnn.py
@@ -37,8 +37,8 @@ def validate(data_type, model, seq_length=40, saved_model=None,
print(rm.model.metrics_names)
def main():
- model = 'mlp'
- saved_model = 'data/ucf101/checkpoints/mlp-features.023-0.926.hdf5'
+ model = 'lstm'
+ saved_model = 'data/checkpoints/lstm-features.026-0.239.hdf5'
if model == 'conv_3d' or model == 'lrcn':
data_type = 'images'
@@ -48,7 +48,7 @@ def main():
image_shape = None
validate(data_type, model, saved_model=saved_model,
- image_shape=image_shape)
+ image_shape=image_shape, class_limit=4)
if __name__ == '__main__':
main()
--
GitLab