Skip to content
Snippets Groups Projects
Commit 571dc432 authored by tomrink's avatar tomrink
Browse files

minor...

parent 755a5203
Branches
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
modules/deeplearning/icing_cnn.py
+ 6
− 1
View file @ 571dc432
...@@ -16,8 +16,13 @@ CACHE_DATA_IN_MEM = False ...@@ -16,8 +16,13 @@ CACHE_DATA_IN_MEM = False
PROC_BATCH_SIZE = 4096 PROC_BATCH_SIZE = 4096
PROC_BATCH_BUFFER_SIZE = 50000 PROC_BATCH_BUFFER_SIZE = 50000
NumClasses = 2 NumClasses = 2
NumLogits = 1 if NumClasses == 2:
NumLogits = 1
else:
NumLogits = NumClasses
BATCH_SIZE = 128 BATCH_SIZE = 128
NUM_EPOCHS = 100 NUM_EPOCHS = 100
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment