salesforce/Merlion

[FEATURE REQUEST] Can Merlion use GPU to train models?

adminblackjack opened this issue · 2 comments

adminblackjack commented

Can Merlion use GPU to train models?

aadyotb commented

Sorry for the delayed response. Currently, this is on a per-model basis. Some of the deep learning models have a cuda / use_cuda / gpu option which allows them to run on GPU. However, it's currently not done in a unified way. We do plan to make this interface more consistent in the future.

adminblackjack commented

Sorry for the delayed response. Currently, this is on a per-model basis. Some of the deep learning models have a cuda / use_cuda / gpu option which allows them to run on GPU. However, it's currently not done in a unified way. We do plan to make this interface more consistent in the future.

Thank you! I'm looking forwark to it!