Thanks a lot for your reply! That solves the problem.
I thought register_buffer makes weight_fb carried to the device when the network is carried to the device. So register_buffer only makes weight_fb appeared in state_dict and can be saved? Does that mean I don’t need to register it if I don’t care about its value?