error with shapes when loading model 39999.npy

[gaelhugo]

Hello, I can successfully load the model 289999.npy, but when I tried with the model 39999.npy provided here, I've got a shape error:

ValueError: Dimension 0 in both shapes must be equal, but are 1536 and 1024. Shapes are [1536,2048] and [1024,2048]. for 'Assign_2' (op: 'Assign') with input shapes: [1536,2048], [1024,2048].

Is there a workaround to make that model work too ?