Modern_Hopfield_network loading ...