search

NVIDIA / nv-wavenet

Reference implementation of real-time autoregressive wavenet inference
BSD 3-Clause "New" or "Revised" License
735 stars 126 forks source link

FP16 Optimizer in pyTorch implementation #33

Open yhgon opened 6 years ago

yhgon commented 6 years ago

PyTorch Implementation of NV-WaveNet could not utilize FP16 TensorCore in Volta during training. need FP16 Optimizer from APEX