for me it does not work at all I even tried your example code, without profiler do

you need to import the function at first <div class="highlight highlight-source-py

still does not produce any output: <div class="snippet-clipboard-content notransla

try: <div class="highlight highlight-source-python notranslate position-relative o

<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clip

Nor working about pytorch_memlab HOT 4 OPEN

stonesjtu commented on July 17, 2024

Nor working

from pytorch_memlab.

Comments (4)

Stonesjtu commented on July 17, 2024

you need to import the function at first

from pytorch_memlab import profile_every

from pytorch_memlab.

cdcseacave commented on July 17, 2024

still does not produce any output:

import torch
from pytorch_memlab import LineProfiler
from pytorch_memlab import profile_every

@profile_every(1)
def inner():
    torch.nn.Linear(100, 100).cuda()

def outer():
    linear = torch.nn.Linear(100, 100).cuda()
    linear2 = torch.nn.Linear(100, 100).cuda()
    inner()

with LineProfiler(outer, inner) as prof:
    outer()
prof.display()

output:

ubuntu@65:~$ python prof.py
No data collected

from pytorch_memlab.

Stonesjtu commented on July 17, 2024

try:

import torch
from pytorch_memlab import LineProfiler
from pytorch_memlab import profile_every

@profile_every(1)
def inner():
    torch.nn.Linear(100, 100).cuda()

def outer():
    linear = torch.nn.Linear(100, 100).cuda()
    linear2 = torch.nn.Linear(100, 100).cuda()
    inner()

outer()

from pytorch_memlab.

cdcseacave commented on July 17, 2024

ubuntu@65:~$ python prof.py
## inner

active_bytes reserved_bytes line code
         all            all
        peak           peak
      80.00K          2.00M    5 @profile_every(1)
                               6 def inner():
     120.00K          2.00M    7     torch.nn.Linear(100, 100).cuda()

from pytorch_memlab.

Recommend Projects