|
Fix relative import and dataset root call
|
gab-ramos
|
|
2 years ago |
|
Add setup files
|
gab-ramos
|
|
2 years ago |
|
Change print to logging + check with more RAM for working example.
|
gab-ramos
|
|
2 years ago |
|
Remove erronous shape check + typos + metric func signature
|
gab-ramos
|
|
2 years ago |
|
Update dataset output to match Google's FLAX examples.
|
gab-ramos
|
|
2 years ago |
|
Fix labels input shape and implement logits' shape check.
|
gab-ramos
|
|
2 years ago |
|
Implement cross-entropy and dummy metrics
|
gab-ramos
|
|
2 years ago |
|
Modify dataset class to self re-initialize when the epoch is completed.
|
gab-ramos
|
|
2 years ago |
|
Change main script flags + new config elements.
|
gab-ramos
|
|
2 years ago |
|
Add multi-device batch stat sync
|
gab-ramos
|
|
2 years ago |
|
Remove double input normalization. Implement train_state creation.
|
gab-ramos
|
|
2 years ago |
|
Implement function prototypes.
|
gab-ramos
|
|
2 years ago |
|
Add helper training modules to hold utility functions.
|
gab-ramos
|
|
2 years ago |
|
Implement training step function.
|
gab-ramos
|
|
2 years ago |
|
Start training loop
|
gab-ramos
|
|
2 years ago |
|
Change VGG activation saving position to after MaxPool.
|
gab-ramos
|
|
2 years ago |
|
Working decoder structure - shapes
|
gab-ramos
|
|
2 years ago |
|
Implement decoder and classifier structures.
|
gab-ramos
|
|
2 years ago |
|
Modify GLONet output
|
gab-ramos
|
|
2 years ago |
|
Add GLONet FLAX exploration notebook
|
gab-ramos
|
|
2 years ago |
|
Implement batch_norm function.
|
gab-ramos
|
|
2 years ago |
|
Un-nest deconv block function.
|
gab-ramos
|
|
2 years ago |
|
Add BatchNorm to GLONet.
|
gab-ramos
|
|
2 years ago |
|
Remove .transpose() from data loading.
|
gab-ramos
|
|
2 years ago |
|
Import VGG and start implementing GLONet.
|
gab-ramos
|
|
2 years ago |
|
Wrap dataset generator logit in a class.
|
gab-ramos
|
|
2 years ago |
|
Implement data batchifying and data handler notebook.
|
gab-ramos
|
|
2 years ago |
|
Implement data getter and generator
|
gab-ramos
|
|
2 years ago |
|
Add dataset utility funcs.
|
gab-ramos
|
|
2 years ago |
|
Reorganize program flags.
|
gab-ramos
|
|
2 years ago |