-
Notifications
You must be signed in to change notification settings - Fork 3.6k
Description
🚀 Feature
support profiling and hooks from pl17 in pl 1.6, in particular hooks/steps for creating a batch (fetch_batch ) and transferring it to the device - I want to be able to get time for these steps when profiling.
Motivation
due to the fact that in PL 1.7 there is no support for torch 1.8.2, I want PL 1.6 to have useful functions that are in PL 1.7
Submission
Get the opportunity to profile and measure the time at the stage of receiving the batch and transferring the batch to the device.
Alternatives
for good, I want it to be without alternatives, and it was simply implemented as it is in 1.7, tk. we support 1.7 but there are conflicts with the torch version -- we want to keep the lightning and torch versions fully compatible.
Additional context
a question about profiling where the missing steps in 1.6 are indicated
https://pytorch-lightning.slack.com/archives/CRBLFHY79/p1659645752234179
support request sticking out 1.8.2 in pl 1.7
#14086
Rays of love for lightning ⚡
cc @Borda @carmocca @kaushikb11 @ninginthecloud @rohitgr7 @nbcsm @guotuofeng