所以我遇到的問題是我想測量 Google Colab 上 iPython 中細胞系的內存消耗和時間，該細胞系也返回一個對象。我正在使用 執(zhí)行此操作memory-profiler，因此我可以將基準測試的所有結果自動存儲在文件中。我發(fā)現(xiàn)這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)理想情況下，我會在一個訓練函數(shù)調用中分配時間、內存和對象的所有 3 個變量，以節(jié)省訓練時間。
1 回答
揚帆大魚
TA貢獻1799條經(jīng)驗 獲得超9個贊
最后，我使用權重和偏差來同時跟蹤 GPU 和內存消耗。
設置非常簡單，但是我還沒有找到如何提取最大值、最小值等信息。
設置：
import?wandb import?pmdarima?as?pm wandb.init()?? model=?pm.auto_arima(df["y"],?seasonal=True,?m=12,?maxiter=10,?njobs?=?-1)
然后在網(wǎng)站上，您可以看到運行一段時間內的消耗情況
事實證明，您可以通過 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 對象的屬性
添加回答
舉報
0/150
提交
取消