diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index 5c6f30c91afdd7da113003a66956e5c57a035f20..065858e87ca786d3af9a2dfac7593d0a04e2d194 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -22,7 +22,7 @@ if NumClasses == 2: else: NumLogits = NumClasses -BATCH_SIZE = 128 +BATCH_SIZE = 64 NUM_EPOCHS = 60 TRACK_MOVING_AVERAGE = False