Saurabh Sharma comments

Repositories
Issues
Comments

Results 1 comments of


                                            Saurabh Sharma

[BUG] Temporal Fusion Transformer trained on GPU then loaded on CPU does not propagate map_location into the loss metrics as expected

@arizzuto can you share the code piece for loading on rank 0 and forcing it to CPU? Would be helpful if you could show if you are dumping the model...