mm
/
citybert
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

bigger batch size

main
mm 2 years ago
parent
948c337ec2
commit
03313d3904
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      train.py

2
train.py
View File

@ -55,7 +55,7 @@ train_examples, val_examples = train_test_split(
# validation examples can be something like templated sentences # validation examples can be something like templated sentences
# that maintain the same distance as the cities (same context) # that maintain the same distance as the cities (same context)
# should probably add training examples like that too if needed # should probably add training examples like that too if needed
BATCH_SIZE = 48 BATCH_SIZE = 16 * 16
num_examples = len(train_examples) num_examples = len(train_examples)
steps_per_epoch = num_examples // BATCH_SIZE steps_per_epoch = num_examples // BATCH_SIZE

Write Preview
Loading…
Cancel
Save