docs: update colab workers_per_gpu
"from mmcv.runner import set_random_seed\n",
"\n",
"# Modify dataset type and path\n",
- "cfg.dataset_type = 'AdobeComp1kDataset' # the AdobeComp1kDataset is reusable\n",
- "cfg.data_root = './data/alphamatting/'\n",
- "\n",
"cfg.data.train.type = 'AdobeComp1kDataset'\n",
"cfg.data.train.ann_file = './data/alphamatting/training_list.json'\n",
"cfg.data.train.data_prefix = './data/alphamatting/'\n",
"\n",
"# Use smaller batch size for training\n",
"cfg.data.samples_per_gpu = 4\n",
- "cfg.data.workers_per_gpu = 4\n",
+ "cfg.data.workers_per_gpu = 2\n",
"\n",
"# The original learning rate (LR) is set for batch size 16 with 1 GPU.\n",
"# We reduce the lr by a factor of 4 since we reduce the batch size.\n",
