afem/checkpoints_beforedwr/model_final.pt