AttributeError: Can't get attribute 'train' on <module '__main__' (built-in)>

wandb --version && python --version && uname

• Weights and Biases version: wandb, version 0.8.31
• Python version: Python 3.6.7
• Operating System: Windows 10

Description

Running tutorial example fails. Link to the tutorial: https://colab.research.google.com/drive/181GCGp36_75C2zm7WLxr9U2QjMXXoibt

What I Did

Running tutorial example fails. Link to the tutorial: https://colab.research.google.com/drive/181GCGp36_75C2zm7WLxr9U2QjMXXoibt

The line with wandb.agent(sweep_id, train)

Gives an error:

Traceback (most recent call last):
File "<string>", line 1, in <module>
File "c:\users\Admin\appdata\local\programs\python\python36\lib\multiprocessing\spawn.py", line 105, in spawn_main exitcode = _main(fd)
File "c:\users\Admin\appdata\local\programs\python\python36\lib\multiprocessing\spawn.py", line 115, in _main self = reduction.pickle.load(from_parent)
AttributeError: Can't get attribute 'train' on <module '__main__' (built-in)>

[nhabbash]

I’ve encountered the same error using:

• wandb, version 0.8.32
• Python 3.7.4 (Anaconda)
• Windows 10

I found out that the issue was with Jupyter Notebook (what I was using) not fully supporting multiprocessing. Haven’t dug out the details, but a fix that made sweeps work was putting the train function definition in an outside .py file, and after that importing it into the notebook.

[vanpelt]

Hey @Pavlov228 this is due to the train function not being defined. Did you run the cell that defined train before calling wandb.agent?