Skip to content

Commit 259970a

Browse files
committed
Fix error message
1 parent 664824b commit 259970a

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

torch_lr_finder/lr_finder.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,9 @@ def dataset(self):
3131
def inputs_labels_from_batch(self, batch_data):
3232
if not isinstance(batch_data, list) and not isinstance(batch_data, tuple):
3333
raise ValueError(
34-
"Your batch type not supported: {}. Please inherit from "
35-
"`TrainDataLoaderIter` (or `ValDataLoaderIter`) and redefine "
36-
"`_batch_make_inputs_labels` method.".format(type(batch_data))
34+
"Your batch type is not supported: {}. Please inherit from "
35+
"`TrainDataLoaderIter` or `ValDataLoaderIter` and override the "
36+
"`inputs_labels_from_batch` method.".format(type(batch_data))
3737
)
3838

3939
inputs, labels, *_ = batch_data
@@ -87,6 +87,7 @@ class ValDataLoaderIter(DataLoaderIter):
8787
loader_iter = iter(loader_iter) # __iter__ is called by `iter()`
8888
```
8989
"""
90+
9091
def __init__(self, data_loader):
9192
super().__init__(data_loader)
9293
self.run_limit = len(self.data_loader)

0 commit comments

Comments
 (0)