search

Accelergy-Project / accelergy

Accelergy is an energy estimation infrastructure for accelerator energy estimations
http://accelergy.mit.edu
MIT License
124 stars 40 forks source link

Support for 3D CNN #8

Closed Diksha-Moolchandani closed 3 years ago

Diksha-Moolchandani commented 3 years ago

Hi,

Do you have some support for 3D CNNs in accelergy?

thanks

nellie-wu commented 3 years ago

Accelergy is only responsible for energy estimation of the components in the architecture, not the workload. If you are using timeloop-accelergy infrastructure, timeloop is the tool that takes in the the workload shapes, which are directly related to the dimension of your input DNN model.

An example of 3D conv specification for Timeloop can be found here: https://github.com/NVlabs/timeloop/issues/64

Diksha-Moolchandani commented 3 years ago

Thanks