Does the official checkpoint "epoch=399-step=23200.ckpt" of the virtual stainng model VSCyto2D supported for fine-tune?

[chooooock]

Thanks for the outstanding works!
On the quick_start.py we can see that this checkpoint realted to the model of FcmaeUNet(from viscy\translation\engine.py). When I try to fine-tune this checkpoint with the GPUTransformDataModule and CombinedDataModule( the data requirements described in the sourcecode in FcmaeUNet), I have to modify the sourcecode of the HCSDataModule(form viscy\data\hcs.py).
I still comes with some error about data matching even though I have changed the HCSDataModule strucutre.
Does this checkpoint was supported for fine-tune? Are there any script for this task?

[edyoshikun]

yes, you should be able to fine tune. You just have to make sure you have the right architecture. I dont know if VSCyto2D supports the GPUTransforms yet as one would have to modify the __get_items__ in the HCSDataModule. Do you mind either opening a PR or sharing where you have modified the source code so I can help a bit more?