Spaces:
Sleeping
Sleeping
| _base_ = [ | |
| 'svtr-tiny_20e_st_mj.py', | |
| ] | |
| model = dict( | |
| preprocessor=dict(output_image_size=(48, 160), ), | |
| encoder=dict( | |
| img_size=[48, 160], | |
| max_seq_len=40, | |
| out_channels=256, | |
| embed_dims=[128, 256, 384], | |
| depth=[3, 6, 9], | |
| num_heads=[4, 8, 12], | |
| mixer_types=['Local'] * 8 + ['Global'] * 10), | |
| decoder=dict(in_channels=256)) | |
| train_dataloader = dict(batch_size=256, ) | |