|
|
|
|
|
by d3m0t3p
920 days ago
|
|
One of the easiest approache is torch.compile, it's the latest iteration of pytorch compiler (previous methods were : TorchScript and FX Tracing.) You simply write
model = torch.compile(model) "Across these 163 open-source models torch.compile works 93% of time, and the model runs 43% faster in training on an NVIDIA A100 GPU. At Float32 precision, it runs 21% faster on average and at AMP Precision it runs 51% faster on average."[1] What google is trying to do, is to involve more people in the R&D of these kind of methods. [1]https://pytorch.org/get-started/pytorch-2.0/ |
|
It actually sounds very useful and cool, I just completely did not get that from the article.