TorchDSP / torchsig

TorchSig is an open-source signal processing machine learning toolkit based on the PyTorch data handling pipeline.
MIT License
155 stars 37 forks source link

Invalid argument for DatasetCreator().__init__() #235

Closed IsaiahHarvi closed 6 days ago

IsaiahHarvi commented 2 months ago

When running generate_wideband_sig53.py, a type error is raised when initializing DatasetCreator with the num_workers argument.

The issued command: python3 generate_wideband_sig53.py --root=data/sig53_wb --all=True

Response:

generate_wideband_sig53.py", line 33, in generate
    creator = DatasetCreator(
              ^^^^^^^^^^^^^^^
TypeError: DatasetCreator.__init__() got an unexpected keyword argument 'num_workers'

After removing that argument and rerunning, the dataset began to generate properly.

Ubuntu, Torchsig v0.4.1

TorchDSP commented 6 days ago

These issues have been resolved in updated main.