snap-stanford / GEARS

GEARS is a geometric deep learning model that predicts outcomes of novel multi-gene perturbations
MIT License
189 stars 38 forks source link

Very long import time #34

Closed dzyim closed 10 months ago

dzyim commented 10 months ago

The package gears takes too much time to import. (__version__ = '0.0.3')

$time python -c 'from gears import PertData'

real    1m37.423s
user    0m14.010s
sys     0m5.412s
$time python -c 'import torch'

real    0m18.923s
user    0m2.117s
sys     0m1.732s
yhr91 commented 10 months ago

Thanks for sharing. It doesn't seem significantly long but please let us know if you have suggestions to speed up the import