mmpretrain1 MMpretrain custom dataset Train Base Line code _Tutorial 튜토리얼 #mmpretrain_base_ = [ '../_base_/models/mae_vit-base-p16.py', '../_base_/datasets/imagenet_bs512_mae.py', '../_base_/default_runtime.py',]train_dataloader = dict( dataset=dict( type='CustomDataset', data_root='data/custom_dataset/', ann_file='', # We assume you are using the sub-folder format without ann_file data_prefix='', # The `data_root` is t.. 2024. 9. 20. 이전 1 다음 반응형