add dataset and dataloader for sample points (#195)

* add dataset and dataloader for sample points * unittests
2023-11-07 11:34:44 +01:00
parent cd5bc9a558
commit d654259428
19 changed files with 581 additions and 196 deletions
--- a/tests/test_solvers/test_garom.py
+++ b/tests/test_solvers/test_garom.py
@@ -134,7 +134,7 @@ def test_train_cpu():
                                                hidden_dimension=64)
                )

-    trainer = Trainer(solver=solver, max_epochs=4, accelerator='cpu')
+    trainer = Trainer(solver=solver, max_epochs=4, accelerator='cpu', batch_size=20)
    trainer.train()

 def test_sample():