Added inference in python

intel / tiny-dpcpp-nn

SYCL implementation of Fused MLPs for Intel GPUs

BSD 3-Clause "New" or "Revised" License

36 stars 9 forks source link

Closed Yunaik closed 2 months ago

Yunaik commented 2 months ago

Modules class in modules.py now uses inference when model.eval() instead of forward
Refactored benchmark_pytorch to loop over widths