is_smp_segformer_state_dict

tiatoolbox . models . architecture . segformer_hf . is_smp_segformer_state_dict

is_smp_segformer_state_dict(state_dict)

Return True if state_dict uses TIA/SMP SegFormer key names.

Parameters:

state_dict (Mapping[str, torch.Tensor])

Return type:

bool