Time series Timeseries Deep Learning Machine Learning Python Pytorch fastai | State-of-the-art Deep Learning library for Time Series and Sequences in Pytorch / fastai
How can tsai train on muti device? I have four gpus, but I do not know how to use.
reg = TSRegressor(X, y, path='./TSTPlus_models', arch="TSTPlus",device="cuda:2", metrics=rmse, verbose=True)
How can tsai train on muti device? I have four gpus, but I do not know how to use. reg = TSRegressor(X, y, path='./TSTPlus_models', arch="TSTPlus",device="cuda:2", metrics=rmse, verbose=True)