TypeError: DataLoaderIterWrapper object is not an iterator ?

class DataLoaderIterWrapper(object):
     def __init__(self, data_loader, auto_reset=True):
        self.data_loader = data_loader
        self.auto_reset = auto_reset
        self._iterator = iter(data_loader)

    def __next__(self):
        # Get a new set of inputs and labels
        try:
            # inputs, labels, *_ = next(self._iterator)
            inputs, labels = next(self._iterator)
        except StopIteration:
            if not self.auto_reset:
                raise
            self._iterator = iter(self.data_loader)
            # inputs, labels, *_ = next(self._iterator)
            inputs, labels = next(self._iterator)

        return inputs, labels

[davidtvs]

As @NaleRaphael said we dropped support for Py2k recently. However, the last release was left in a broken state for Py2k so it’s only fair that it gets fixed. I released version 0.1.5 which fixes the extended iterable unpacking meaning that you shouldn’t have to comment-out those lines and further change the code to make it work.

Give it a try, it’s currently the latest version on PyPI so you can simply do pip install torch-lr-finder.

I also created a branch with the latest code that works on Py2k: torch_lr_finder-v0.1.

Let us know if you have any problems with the latest version

[davidtvs]

Closing this, feel free to reopen if you find a problem related to this