[Fix]: add cfg.device in colab tutorial to compat the most recent mmdet (#9112)

3 years ago · a12be714e9
parent f70164d2c6
commit a12be714e9
1 changed files with 1 additions and 0 deletions
--- a/demo/MMDet_Tutorial.ipynb
+++ b/demo/MMDet_Tutorial.ipynb
@ -1509,6 +1509,7 @@
        "# Set seed thus the results are more reproducible\n",
        "cfg.seed = 0\n",
        "set_random_seed(0, deterministic=False)\n",
+        "cfg.device = 'cuda'\n",
        "cfg.gpu_ids = range(1)\n",
        "\n",
        "# We can also use tensorboard to log the training process\n",