python - wandb 循环在 Jupyter Notebook 和 Microsoft Visual Studio Code (Pyhton3) 之间的工作方式不同

Question

以下代码应循环 3 次。它在 Jupyter notbook 中工作了 3 次，但在 VSC 中只工作了一次。诡异的。

import wandb
N_EPOCHS=10
batch_size=16
from tqdm import tqdm 
for eachtime in tqdm(range(3)):
     
    wandb.init(project='test',name='TESTWANDB'+'_'+str(eachtime))
    wandb.config = {
                "epochs": N_EPOCHS,
                "batch_size": batch_size
                }

Answer 1

如果您希望每次init调用都创建一个新的运行，您应该调用wandb.finish()并且这将显式结束该运行，然后您可以wandb.init再次调用以开始新的运行。