Searched defs:load_hook (Results 1 – 2 of 2) sorted by relevance
203 def load_hook(self, state_dict, prefix, *args): member in Attention
182 def load_hook(self, state_dict, prefix, *args): member in Attention