Did you call validation_step during training? about anydoor

[XavierCHEN34]

1. Validation: No, we did not call validation, instead we infer several training samples to see the converging process.
2. Picking the best model by observing the lowest loss is not reliable, and neither is calculating some quantitative metrics. So we just save models with different steps.
3. You could repeat the image_data, it is just a simple way to adjust the ratio of different dataset, feel free to use your own implementation.

[ai1361720220000]

In the run_train_anydoor.py it seems the training process will not save any model. I'm confused if i should add pl.callbacks.ModelCheckpoint in the callbacks because i'm also not familiar with pytorch lightning..

[Dumeowmeow]

In the run_train_anydoor.py it seems the training process will not save any model. I'm confused if i should add pl.callbacks.ModelCheckpoint in the callbacks because i'm also not familiar with pytorch lightning..

Hi, have you solved it now?I Iran into the same problem and couldn't find where to save the model after training.