search

AutoResearch / EEG-GAN

Other
19 stars 1 forks source link

Autoencoder says GAN in places #64

Closed chadcwilliams closed 3 months ago

chadcwilliams commented 8 months ago

In the parameters and when saving the model.

whyhardt commented 3 months ago

Fixed in system_inputs.py

Could not reproduce for saving the model:

-----------------------------------------
Command line arguments:
-----------------------------------------

Number of epochs: 1
-----------------------------------------

Training the first level of the autoencoder...
100%|██████████████████████████████████████████████████████████| 1/1 [00:00<00:00,  2.29it/s, TRAIN LOSS: 0.018711, TEST LOSS: 0.012502]
Managing checkpoints...
Training the second level of the autoencoder...
100%|██████████████████████████████████████████████████████████| 1/1 [00:00<00:00,  1.48it/s, TRAIN LOSS: 0.018746, TEST LOSS: 0.013592]
Managing checkpoints...
Training finished.
Model and configuration saved in trained_ae/ae_gansEEGTrainingData_1713259748.pt