From df052771e6abcca2c6c810c8e1bbe64a6f275d82 Mon Sep 17 00:00:00 2001 From: YuanLiuuuuuu <3463423099@qq.com> Date: Thu, 14 Apr 2022 19:58:34 +0800 Subject: [PATCH] [Fix]: Change imgs_per_gpu to samples_per_gpu MAE --- configs/selfsup/_base_/datasets/imagenet_mae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/configs/selfsup/_base_/datasets/imagenet_mae.py b/configs/selfsup/_base_/datasets/imagenet_mae.py index 939fc1039..cd833b5ab 100644 --- a/configs/selfsup/_base_/datasets/imagenet_mae.py +++ b/configs/selfsup/_base_/datasets/imagenet_mae.py @@ -17,7 +17,7 @@ # dataset summary data = dict( - imgs_per_gpu=128, + samples_per_gpu=128, workers_per_gpu=8, train=dict( type=dataset_type,