Set sb3_imitation.py to CPU to fix error with CUDA#196

Ivan-267 merged 2 commits intomainedbeeching/godot_rl_agents:mainfrom

Ivan-267-set-imitation-to-cpu-deviceedbeeching/godot_rl_agents:Ivan-267-set-imitation-to-cpu-deviceCopy head branch name to clipboard