File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -31,9 +31,9 @@ def dataset(self):
31
31
def inputs_labels_from_batch (self , batch_data ):
32
32
if not isinstance (batch_data , list ) and not isinstance (batch_data , tuple ):
33
33
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 ))
37
37
)
38
38
39
39
inputs , labels , * _ = batch_data
@@ -87,6 +87,7 @@ class ValDataLoaderIter(DataLoaderIter):
87
87
loader_iter = iter(loader_iter) # __iter__ is called by `iter()`
88
88
```
89
89
"""
90
+
90
91
def __init__ (self , data_loader ):
91
92
super ().__init__ (data_loader )
92
93
self .run_limit = len (self .data_loader )
You can’t perform that action at this time.
0 commit comments