Skip to content

With Pytorch NGC Container - Pull Tag - docker pull nvcr.io/nvidia/pytorch:20.07-py3

License

Notifications You must be signed in to change notification settings

codesteller/time-series-with-transformer

Repository files navigation

Time Series Forecasting with Transformers

With Pytorch NGC Container - Pull Tag - docker pull nvcr.io/nvidia/pytorch:20.07-py3

Multiple GPU Training

python transformer-multigpu.py --gpu_devices 0 1 --batch_size 200

Single GPU Training

python transformer-multigpu.py --gpu_devices 0 --batch_size 200

About

With Pytorch NGC Container - Pull Tag - docker pull nvcr.io/nvidia/pytorch:20.07-py3

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published