To test event_vae

[thomhert]

Hi, Thank you for sharing your work. I successfully trained the vae model using the sample data gates.txt. Now I want to test my trained model using your test.py. Can you describe how to use the test.py? Thank you!

[saihv]

@thomhert Testing the event VAE will be pretty similar to the way the train script is set up, you just need to load a test set of events, and then do a forward pass with the model.

I have updated the test script in the event_vae folder to help with this. Right now, the test script loads a trained eVAE model, a stream of events, then samples batches of events with increasing length and computes the reconstructions for each value of length. This replicates the 'sparsity invariance' experiment from our paper (Figure ). Hopefully this helps as a starting point for testing trained VAE models!