search

lhnguyen102 / cuTAGI

CUDA implementation of Tractable Approximate Gaussian Inference
MIT License
29 stars 9 forks source link

Update TimeSeriesDataloader #75

Closed davidwardan closed 1 month ago

davidwardan commented 1 month ago

Commented two lines in batch generator to fix the bug where the model does not run when the batch size is smaller than num_data

lhnguyen102 commented 1 month ago

@davidwardan The target is correct this time. Let's wait for the checks to pass before merging