Save inference images
xiaofeng-c opened this issue · comments
If i want to save the inference images to the folder given,waht should i do?
Hi, if you want to save the images during training you can just get the images in the validation_step. Otherwise, if you just want to do inference with a pretrained model you can just load the model and do test(), see PyTorchLightning docs
OK,i get it,thank you!