search

Lightning-AI / pytorch-lightning

Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
https://lightning.ai
Apache License 2.0
28.52k stars 3.39k forks source link

TPU profiler #18521

Open ethanhe42 opened 1 year ago

ethanhe42 commented 1 year ago

📚 Documentation

is TPU profiler doc up to date? https://lightning.ai/docs/pytorch/latest/tuning/profiler_advanced.html

I don't see the profile page in tensorboard after following the steps

cc @borda @carmocca @nbcsm @guotuofeng

Bhavay-2001 commented 1 year ago

Hi @awaelchli, I'm a beginner in open source and want to start contributing. Can you please assist me on how to solve this issue? I would really like to help.

awaelchli commented 1 year ago

@Bhavay-2001 The docs here would need a refresh. It requires someone who is familiar with TPU training that can run and test the profiler on the updated tutorial. If you aren't familiar with TPU training, I would suggest to pick a different topic to get started.