So, I was trying to make the Vegas into a tf.function. It is relatively easy actually but I realised we don't necessarily want to do that. If we do we will create a graph with n_events node, which is extremely big.
Otherwise we can do the batch thingy, but the problem of doing a graph where the number of events is something that come from the outside is that if two different numbers of events go in it will trigger two different compilations.
tl;dr can be done, I'll try uploading it before the lecture is finished, but I don't think we want it as a default
So, I was trying to make the Vegas into a tf.function. It is relatively easy actually but I realised we don't necessarily want to do that. If we do we will create a graph with
n_eventsnode, which is extremely big. Otherwise we can do the batch thingy, but the problem of doing a graph where the number of events is something that come from the outside is that if two different numbers of events go in it will trigger two different compilations.tl;dr can be done, I'll try uploading it before the lecture is finished, but I don't think we want it as a default