Perf: replace unnecessary torch.split
with indexing (#4505)
#63
Job | Run time |
---|---|
21m 9s | |
1m 3s | |
2m 29s | |
24m 41s |
torch.split
with indexing (#4505)
#63
Job | Run time |
---|---|
21m 9s | |
1m 3s | |
2m 29s | |
24m 41s |