Support for FILM SIREN in odak.learn.models.multi_layer_perceptron.

kaanaksit · Nov 30, 2023 · e47c5eb · e47c5eb
1 parent 1ab617e
commit e47c5eb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_learn_models_multi_layer_perceptron.py b/test/test_learn_models_multi_layer_perceptron.py
@@ -11,10 +11,10 @@ def main():
     test_filename  = './estimation.png'
     weights_filename = 'model_weights.pt'
     learning_rate = 1e-3
-    no_epochs = 2000
+    no_epochs = 20
     number_of_batches = 1
     dimensions = [2, 128, 128, 3]
-    device_name = 'cuda'
+    device_name = 'cpu'
     save_at_every = 500
     device = torch.device(device_name)
     model = odak.learn.models.multi_layer_perceptron(