if you mean a subset of targets, something like that:
d = torch.load(f"FOLD{fold}_.pth")
for suffix in (".weight",".bias"):
  key = PREFIX + suffix
  d[key]=map_402_to_206(d[key])
model.load_state_dict(d)

PREFIX is output linear layer’s key. map_402_to_206 should reflect how you select a subset, simplest case would be a subrange: d[key] = d[key][:206]
if you want to reuse hidden layers on a different set of targets, delete above keys instead and use strict=False