theislab / chemCPA

Code for "Predicting Cellular Responses to Novel Drug Perturbations at a Single-Cell Resolution", NeurIPS 2022.
https://arxiv.org/abs/2204.13545
MIT License
88 stars 23 forks source link

Calc disentanglement loss on GPU #41

Closed siboehm closed 2 years ago

siboehm commented 2 years ago

Very roughly, running 1000 epochs to calc just the perturbation disentanglement took ~20 minutes.