diff --git a/fastNLP/core/batch.py b/fastNLP/core/batch.py index 1a31e92a..b14b21de 100644 --- a/fastNLP/core/batch.py +++ b/fastNLP/core/batch.py @@ -201,6 +201,19 @@ class TorchLoaderIter(BatchIter): self.batch_size = dataset.batch_size +class OnlineDataGettter: + # TODO + pass + + +class OnlineDataIter(BatchIter): + # TODO + def __init__(self, dataset, batch_size=1, buffer_size=10000, sampler=None, as_numpy=False, + num_workers=0, pin_memory=False, drop_last=False, + timeout=0, worker_init_fn=None, **kwargs): + super().__init__() + + def _to_tensor(batch, field_dtype): """