afem/checkpoints_beforedwr/model_iter0400.pt