Can virtual staining for high-throughput screening generalize?