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

bigger batch size

Browse Source
This commit is contained in:
mm 2023-05-05 01:28:35 +00:00
parent 948c337ec2
commit 03313d3904
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

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