File size: 631 Bytes
729c925 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
###########################################################################
# Created by: NTU
# Email: heshuting555@gmail.com
# Copyright (c) 2024
###########################################################################
import pickle
ckpt = pickle.load(open('model_final_86143f.pkl', 'rb'))
ckpt['model']['sem_seg_head.predictor.query_embed.weight'] = ckpt['model']['sem_seg_head.predictor.query_embed.weight'][:20]
ckpt['model']['sem_seg_head.predictor.static_query.weight'] = ckpt['model']['sem_seg_head.predictor.static_query.weight'][:20]
with open("model_final_86143f.pkl", 'wb') as file:
pickle.dump(ckpt, file) |