jansel / pytorch-jit-paritybench Goto Github PK

Hi,

I tried to generate Test Scripts for "649453932_Chinese_Text_Classification_Pytorch.zip" but it failed with Error as Below:
The script is like: python main.py -g ./649453932_Chinese_Text_Classification_Pytorch.zip --tests-dir ./

Is there anything I'm overlooking?

Error from subprocess
Traceback (most recent call last):
File "/home/pytorch/molly/paritytest/pytorch-jit-paritybench/paritybench/utils.py", line 51, in call_with_timeout_subproc
result = fn(*args, *kwargs)
File "/home/pytorch/molly/paritytest/pytorch-jit-paritybench/paritybench/generate.py", line 42, in generate_zipfile_subproc
extractor.main(path)
File "/home/pytorch/molly/paritytest/pytorch-jit-paritybench/paritybench/module_extractor.py", line 308, in main
self.test_modules()
File "/home/pytorch/molly/paritytest/pytorch-jit-paritybench/paritybench/module_extractor.py", line 262, in test_modules
self.test_nn_module(name, value)
File "/home/pytorch/molly/paritytest/pytorch-jit-paritybench/paritybench/module_extractor.py", line 281, in test_nn_module
stats, errors, testcases = call_with_timeout(
File "/home/pytorch/molly/paritytest/pytorch-jit-paritybench/paritybench/utils.py", line 26, in call_with_timeout
proc.start()
File "/home/pytorch/anaconda3/envs/torch20_parity/lib/python3.8/multiprocessing/process.py", line 121, in start
self._popen = self._Popen(self)
File "/home/pytorch/anaconda3/envs/torch20_parity/lib/python3.8/multiprocessing/context.py", line 224, in _Popen
return _default_context.get_context().Process._Popen(process_obj)
File "/home/pytorch/anaconda3/envs/torch20_parity/lib/python3.8/multiprocessing/context.py", line 284, in _Popen
return Popen(process_obj)
File "/home/pytorch/anaconda3/envs/torch20_parity/lib/python3.8/multiprocessing/popen_spawn_posix.py", line 32, in init
super().init(process_obj)
File "/home/pytorch/anaconda3/envs/torch20_parity/lib/python3.8/multiprocessing/popen_fork.py", line 19, in init
self._launch(process_obj)
File "/home/pytorch/anaconda3/envs/torch20_parity/lib/python3.8/multiprocessing/popen_spawn_posix.py", line 47, in _launch
reduction.dump(process_obj, fp)
File "/home/pytorch/anaconda3/envs/torch20_parity/lib/python3.8/multiprocessing/reduction.py", line 60, in dump
ForkingPickler(file, protocol).dump(obj)
_pickle.PicklingError: Can't pickle <class 'paritybench.module_extractor.output.Encoder'>: import of module 'paritybench.module_extractor.output' failed