Starbeamrainbowlabs 6a29105f56 model_segmentation: stack not reshape		2022-10-25 21:25:15 +01:00
..
__init__.py	ai: start creating initial model implementation.	2022-08-10 19:03:25 +01:00
CallbackCustomModelCheckpoint.py	ai: implement saving only the rainfall encoder	2022-09-06 19:48:46 +01:00
convnext.py	model_contrastive: make water encoder significantly shallower	2022-10-24 20:52:31 +01:00
convnext_inverse.py	segmentation: adds xxtiny, but unsure if it's small enough	2022-10-11 19:22:37 +01:00
LayerCheeseMultipleOut.py	prepare for NCE loss	2022-10-25 21:15:05 +01:00
LayerContrastiveEncoder.py	ai Bugfix LayerContrastiveEncoder: channels → input_channels	2022-09-05 23:53:16 +01:00
LayerConvNeXtGamma.py	LayerConvNeXtGamma: avoid adding an EagerTensor to config	2022-10-12 17:12:07 +01:00
LayerStack2Image.py	model_segmentation: stack not reshape	2022-10-25 21:25:15 +01:00
LossContrastive.py	ai: implement saving only the rainfall encoder	2022-09-06 19:48:46 +01:00