The problem of the number of categories in pre-training？

[Yuyiba]

Hi,This is perhaps a very simple question, but I don’t understand
Using my data set, there are 5 categories,I wonder if we need change 19 to 5?
thanks!
new_params = model.state_dict().copy()
for i in saved_state_dict:
# Scale.layer5.conv2d_list.3.weight
i_parts = i.split('.')
# print i_parts
if not args.num_classes == 19 or not i_parts[1] == 'layer5':
new_params['.'.join(i_parts[1:])] = saved_state_dict[i]
# print i_parts
model.load_state_dict(new_params)

[lychrel]

which file is that in?