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

Open adminblackjack opened this issue 3 years ago • 2 comments

Can Merlion use GPU to train models?

Feb 27 '22 09:02 adminblackjack

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.

Mar 29 '22 22:03 aadyotb

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!

Apr 01 '22 00:04 adminblackjack