commit a06147455497e4b36507d32db691f545d9815274
parent 250d5489cb21e1566714382a31c0044af74e8acc
Author: keith <kbloemer89@gmail.com>
Date: Fri, 4 Dec 2020 20:31:56 -0600
Reduced default input_size to 100
Diffstat:
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/train.py b/train.py
@@ -165,6 +165,6 @@ if __name__ == "__main__":
parser.add_argument("--batch_size", type=int, default=4096)
parser.add_argument("--max_epochs", type=int, default=1)
parser.add_argument("--create_plots", type=int, default=1)
- parser.add_argument("--input_size", type=int, default=150)
+ parser.add_argument("--input_size", type=int, default=100)
args = parser.parse_args()
main(args)
\ No newline at end of file