ksanjeevan / crnn-audio-classification

UrbanSound classification using Convolutional Recurrent Networks in PyTorch
MIT License
380 stars 80 forks source link

macOS? #21

Open javadba opened 3 years ago

javadba commented 3 years ago

Can this project run on macOS? If so can some notes be added? I get the following errors (on catalina). It appears to be a problem with loading the model

AttributeError: Can't pickle local object 'AugmentationTransform._get_dist..'

tqdm _trange= 0%| | 0/5 [00:00<?, ?it/s] 0%| | 0/5 [00:00<?, ?it/s]Traceback (most recent call last): File "/Users/steve/git/crnn-audio-classification/run.py", line 194, in train_main(config, args.resume) File "/Users/steve/git/crnn-audio-classification/run.py", line 131, in train_main trainer.train() File "/Users/steve/git/crnn-audio-classification/train/base_trainer.py", line 89, in train result = self._train_epoch(epoch) File "/Users/steve/git/crnn-audio-classification/train/trainer.py", line 62, in _train_epoch for batch_idx, batch in enumerate(_trange): File "/usr/local/lib/python3.8/site-packages/tqdm/std.py", line 1102, in iter for obj in iterable: File "/usr/local/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 352, in iter return self._get_iterator() File "/usr/local/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 294, in _get_iterator return _MultiProcessingDataLoaderIter(self) File "/usr/local/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 801, in init w.start() File "/usr/local/Cellar/python@3.8/3.8.6_2/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/process.py", line 121, in start self._popen = self._Popen(self) File "/usr/local/Cellar/python@3.8/3.8.6_2/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/context.py", line 224, in _Popen return _default_context.get_context().Process._Popen(process_obj) File "/usr/local/Cellar/python@3.8/3.8.6_2/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/context.py", line 284, in _Popen return Popen(process_obj) File "/usr/local/Cellar/python@3.8/3.8.6_2/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/popen_spawn_posix.py", line 32, in init super().init(process_obj) File "/usr/local/Cellar/python@3.8/3.8.6_2/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/popen_fork.py", line 19, in init self._launch(process_obj) File "/usr/local/Cellar/python@3.8/3.8.6_2/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/popen_spawn_posix.py", line 47, in _launch reduction.dump(process_obj, fp) File "/usr/local/Cellar/python@3.8/3.8.6_2/Frameworks/Python.framework/Versions/3.8/lib/python3.8/multiprocessing/reduction.py", line 60, in dump ForkingPickler(file, protocol).dump(obj) AttributeError: Can't pickle local object 'AugmentationTransform._get_dist..'