LINCellularNeuroscience / VAME

Variational Animal Motion Embedding - A tool for time series embedding and clustering

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

vame.train_model() stuck

kritgrover opened this issue · comments

I have created the trainset and initialized all the parameters as needed, but my train_model() is just stuck like this for the past 20 minutes. I am using a RTX 3080 so I don't think this is a bottleneck, but if anyone has any suggestion or explanation, please let me know.

image