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

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

所以我遇到的问题是我想测量内存消耗和Google Colab上iPython中单元格行的时间，该行也在返回对象。 我使用memory-profiler进行此操作，因此可以将基准测试的所有结果自动存储在文件中。

我发现memit -o给了我一个MemitResult对象，但是 问题是对象分配给出了未定义的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个变量， 以节省培训时间。

解决方法

最后，我使用Weights and biases来同时跟踪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对象的属性