Jabrain commited on
Commit
af67342
·
1 Parent(s): 6e97713

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def split_input_target(chunk): # Define a function to split the input and target
26
  dataset = sequences.map(split_input_target) # Apply the function to the dataset
27
 
28
  # Shuffle and batch the dataset
29
- BATCH_SIZE = 64 # Batch size
30
  BUFFER_SIZE = 10000 # Buffer size for shuffling
31
  dataset = dataset.shuffle(BUFFER_SIZE).batch(BATCH_SIZE, drop_remainder=True) # Shuffle and batch the dataset
32
 
 
26
  dataset = sequences.map(split_input_target) # Apply the function to the dataset
27
 
28
  # Shuffle and batch the dataset
29
+ BATCH_SIZE = 1 # Batch size
30
  BUFFER_SIZE = 10000 # Buffer size for shuffling
31
  dataset = dataset.shuffle(BUFFER_SIZE).batch(BATCH_SIZE, drop_remainder=True) # Shuffle and batch the dataset
32