Skip to content

Commit 4dc0894

Browse files
NotImplementedError: Model's DEFAULT_TRAIN_BSIZE is not implemented. (#2563)
Summary: Some tests fail in train mode. I found missing fields in specific models and added DEFAULT_TRAIN_BSIZE with the same value as eval batch size. Pull Request resolved: #2563 Reviewed By: adamomainz Differential Revision: D69786156 Pulled By: xuzhao9 fbshipit-source-id: aedf72d758d31e59f3c9b611d1ae0a46fee7c7f4
1 parent ee4aa8b commit 4dc0894

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torchbenchmark/models/cm3leon_generate/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
class Model(BenchmarkModel):
99
task = NLP.LANGUAGE_MODELING
10+
DEFAULT_TRAIN_BSIZE = 1
1011
DEFAULT_EVAL_BSIZE = 1
1112

1213
def __init__(self, test, device, batch_size=None, extra_args=[]):

0 commit comments

Comments
 (0)