bene-ges

@traidn, maybe this is not an official way, but you can try ``` state_dict = torch.load("model_weights.ckpt", map_location=device) asr_model.load_state_dict(state_dict) ``` you can get `model_weights.ckpt` if you unpack the nemo checkpoint with...

@traidn - maybe you can try to load weights and then train like you did from scratch? Without resuming from checkpoint

Hi, we had the same error after the successful building fast_rnnt for AMD using Rocm 5.4 with correct installed pytorch 2.0.1 and torchaudio 0.15.2 ``` File "/home/ubnt/anaconda3/lib/python3.8/site-packages/fast_rnnt/rnnt_loss.py", line 533, in...

It seems that Rocm isn't supported in the build. `-- No NVCC detected. Disable CUDA support`

@danpovey, rocm can compile CUDA code into the amd binary. Most of projects just add the rocm compile commands like Pytorch does. So the Pytorch build system can be an...

another useful link on porting CUDA (all notations almost identical) https://www.lumi-supercomputer.eu/preparing-codes-for-lumi-converting-cuda-applications-to-hip/

I can help with testing on amd if needed