dmarx's picture
added checkpoint with ema + non-ema weights
9ec8259