search

reczoo / BARS

BARS: Towards Open Benchmarking for Recommender Systems https://openbenchmark.github.io/BARS
Apache License 2.0
350 stars 57 forks source link

jit.trace not working for the models #25

Closed santoshcoder23 closed 9 months ago

santoshcoder23 commented 1 year ago 
torch.jit.trace(fibinet, batch_data)

File "/anaconda3/envs/fuxictr3.8/lib/python3.8/site-packages/torch/jit/_trace.py", line 794, in trace return trace_module( File "/anaconda3/envs/fuxictr3.8/lib/python3.8/site-packages/torch/jit/_trace.py", line 1056, in trace_module module._c._create_method_from_trace( File "/anaconda3/envs/fuxictr3.8/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl return forward_call(*args, *kwargs) File "/anaconda3/envs/fuxictr3.8/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1488, in _slow_forward result = self.forward(input, **kwargs) TypeError: forward() takes 2 positional arguments but 3 were given

zhujiem commented 9 months ago

Sorry that I am not familar with jit.trace