在 IPython 中使用 memit magic 时返回 MemitResult 对象和函数调用

Question

所以我遇到的问题是我想测量 Google Colab 上 iPython 中细胞系的内存消耗和时间，该细胞系也返回一个对象。我正在使用 执行此操作memory-profiler，因此我可以将基准测试的所有结果自动存储在文件中。我发现这memit -o给了我一个MemitResult对象，但问题是对象分配给出了一个 undefined m̀odel：NameError: name 'model' is not defined 简短的 MWE：%%time import pmdarima as pm trainMemory =  %memit  -o  model= pm.auto_arima(df["y"], seasonal=True, m=12, maxiter=10, njobs = -1)理想情况下，我会在一个训练函数调用中分配时间、内存和对象的所有 3 个变量，以节省训练时间。

Answer 1

最后，我使用权重和偏差来同时跟踪 GPU 和内存消耗。

设置非常简单，但是我还没有找到如何提取最大值、最小值等信息。

设置：

import wandb
import pmdarima as pm
wandb.init()  
model= pm.auto_arima(df["y"], seasonal=True, m=12, maxiter=10, njobs = -1)

然后在网站上，您可以看到运行一段时间内的消耗情况

事实证明，您可以通过 api 获得最大使用量，而无需查看图表

import wandb 

api = wandb.Api() 

run = api.run(f"YOURNAME/{wandb.run.name}/{wandb.run.id}") 

system_metrics = run.history(stream = 'events') print(system_metrics["system.gpu.0.memory"].max())

system_metrics.to_csv("metrics.csv")

然后您可以访问 system_metrics 对象的属性