Hi,thanks for your patience.
I'm new here,and when I try to run the train in CUB200,meet the error
Could you please help .THANGKS.
Traceback (most recent call last):
File "main.py", line 403, in
main(config)
File "main.py", line 163, in main
train_one_epoch_local_data(config, model, criterion, data_loader_train, optimizer, epoch, mixup_fn, lr_scheduler)
File "main.py", line 210, in train_one_epoch_local_data
outputs = model(samples,meta)
File "/home/miniconda3/envs/pytorch/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/miniconda3/envs/pytorch/lib/python3.8/site-packages/torch/nn/parallel/distributed.py", line 705, in forward
output = self.module(*inputs[0], **kwargs[0])
File "/home/miniconda3/envs/pytorch/lib/python3.8/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
result = self.forward(*input, **kwargs)
File "/home/MetaFormer/models/MetaFG_meta.py", line 231, in forward
x = self.forward_features(x,meta)
File "/home/MetaFormer/models/MetaFG_meta.py", line 171, in forward_features
metas = torch.split(meta,self.meta_dims,dim=1)
File "/home/miniconda3/envs/pytorch/lib/python3.8/site-packages/torch/functional.py", line 156, in split
return tensor.split(split_size_or_sections, dim)
File "/home/miniconda3/envs/pytorch/lib/python3.8/site-packages/torch/tensor.py", line 499, in split
return super(Tensor, self).split_with_sizes(split_size, dim)
RuntimeError: split_with_sizes expects split_sizes to sum exactly to 32 (input tensor's size at dimension 1), but got split_sizes=[4, 3]