Add DeMemWM curriculum training support

Fix DeMemWM checkpoint load priority

Keep reproduction train and eval fixes

update

update evaluation and fix psnr

first commit