afem/checkpoints_beforedwr/model_iter0401.pt