Commit graph

210 commits

Author SHA1 Message Date
Starbeamrainbowlabs fc6d2dabc9
Upscale first, THEN convnext... 2022-10-03 16:38:43 +01:00
Starbeamrainbowlabs 6a0790ff50
convnext_inverse: add returns; change ordering 2022-10-03 16:32:09 +01:00
Starbeamrainbowlabs fe813cb46d
slurm predict: fix plotting subcall 2022-10-03 16:03:26 +01:00
Starbeamrainbowlabs e51087d0a9
add reshape layer 2022-09-28 18:22:48 +01:00
Starbeamrainbowlabs a336cdee90
and continues 2022-09-28 18:18:10 +01:00
Starbeamrainbowlabs de47a883d9
missing units 2022-09-28 18:17:22 +01:00
Starbeamrainbowlabs b5e08f92fe
the long night continues 2022-09-28 18:14:09 +01:00
Starbeamrainbowlabs dc159ecfdb
and again 2022-09-28 18:11:46 +01:00
Starbeamrainbowlabs 4cf0485e32
fixup... again 2022-09-28 18:10:11 +01:00
Starbeamrainbowlabs 030d8710b6
fixup 2022-09-28 18:08:31 +01:00
Starbeamrainbowlabs 4ee7f2a0d6
add water thresholding 2022-09-28 18:07:26 +01:00
Starbeamrainbowlabs 404dc30f08
and again 2022-09-28 17:39:09 +01:00
Starbeamrainbowlabs 4cd8fc6ded
segmentation: param name fix 2022-09-28 17:37:42 +01:00
Starbeamrainbowlabs 41ba980d69
segmentationP implement dataset parser 2022-09-28 17:19:21 +01:00
Starbeamrainbowlabs d618e6f8d7
pretrain-predict: params.json → metadata.jsonl 2022-09-28 16:35:22 +01:00
Starbeamrainbowlabs e9e6139c7a
typo 2022-09-28 16:28:18 +01:00
Starbeamrainbowlabs 3dee3d8908
update cli help 2022-09-28 16:23:47 +01:00
Starbeamrainbowlabs b836f7f70c
again 2022-09-27 19:06:41 +01:00
Starbeamrainbowlabs 52dff130dd
slurm pretraian predict: moar logging 2022-09-27 19:05:54 +01:00
Starbeamrainbowlabs 2d24174e0a
slurm pretrain predict: add $OUTPUT 2022-09-27 18:58:02 +01:00
Starbeamrainbowlabs 7d0e3913ae
fix logging 2022-09-27 18:51:58 +01:00
Starbeamrainbowlabs d765b3b14e
fix crash 2022-09-27 18:43:43 +01:00
Starbeamrainbowlabs 2cd59a01a5
slurm-pretrain-plot: add ARGS 2022-09-27 18:41:35 +01:00
Starbeamrainbowlabs f4d1d1d77e
just wh 2022-09-27 18:25:45 +01:00
Starbeamrainbowlabs 4c24d69ae6
$d → +d 2022-09-27 18:17:07 +01:00
Starbeamrainbowlabs cdb19b4d9f
fixup 2022-09-27 18:13:21 +01:00
Starbeamrainbowlabs c4d3c16873
add some logging 2022-09-27 18:10:58 +01:00
Starbeamrainbowlabs 3772c3227e
fixup 2022-09-27 17:57:21 +01:00
Starbeamrainbowlabs dbfa45a016
write params.json properly 2022-09-27 17:49:54 +01:00
Starbeamrainbowlabs a5455dc22a
fix import 2022-09-27 17:41:24 +01:00
Starbeamrainbowlabs d6ff3fb2ce
pretrain_predict fix write mode 2022-09-27 17:38:12 +01:00
Starbeamrainbowlabs f95fd8f9e4
pretrain-predict: add .tfrecord output function 2022-09-27 16:59:31 +01:00
Starbeamrainbowlabs 30b8dd063e
fixup 2022-09-27 15:54:37 +01:00
Starbeamrainbowlabs 3cf99587e4
Contraster: add gamma layer to load_model 2022-09-27 15:53:52 +01:00
Starbeamrainbowlabs 9e9852d066
UMAP: 100k random
no labels.
2022-09-27 15:52:45 +01:00
Starbeamrainbowlabs 58c65bdc86
slurm: allow predictions on gpu 2022-09-23 19:21:57 +01:00
Starbeamrainbowlabs d59de41ebb
embeddings: change title rendering; make even moar widererer
We need to see that parallel coordinates  plot in detail
2022-09-23 18:56:39 +01:00
Starbeamrainbowlabs b4ddb24589
slurm plot: compute → highmem 2022-09-22 18:27:58 +01:00
Starbeamrainbowlabs df12470e78
flip conda and time
hopefully we can capture the exit code this way
2022-09-21 14:40:28 +01:00
Starbeamrainbowlabs 5252a81238
vis: don't call add_subplot 2022-09-20 19:06:21 +01:00
Starbeamrainbowlabs 32bb55652b
slurm predict: autoqueue UMAP plot 2022-09-16 19:36:57 +01:00
Starbeamrainbowlabs 24c5263bf0
slurm predict-plot: fixup 2022-09-16 19:27:26 +01:00
Starbeamrainbowlabs 7e9c119b04
slurm: fix job name 2022-09-16 19:20:59 +01:00
Starbeamrainbowlabs 1574529704
slurm pretrain-predict: move to compute; make exclusive just in case
also shortent o 3 days
2022-09-16 19:17:42 +01:00
Starbeamrainbowlabs d7f5958af0
slurm: write new job files 2022-09-16 19:00:43 +01:00
Starbeamrainbowlabs a552cc4dad
ai vis: make parallel coordinates wider 2022-09-16 18:51:49 +01:00
Starbeamrainbowlabs a70794e661
umap: no min_dist 2022-09-16 17:09:09 +01:00
Starbeamrainbowlabs 5778fc51f7
embeddings: fix title; remove colourmap 2022-09-16 17:08:04 +01:00
Starbeamrainbowlabs 4fd852d782
fixup 2022-09-16 16:44:35 +01:00
Starbeamrainbowlabs fcab227f6a
cheese: set label for everything to 1 2022-09-16 16:42:05 +01:00
Starbeamrainbowlabs b31645bd5d
pretrain-plot: fix crash; remove water code
the model doesn't save the water encoder at this time
2022-09-16 16:24:07 +01:00
Starbeamrainbowlabs a5f03390ef
pretrain-plot: handle open w -> r 2022-09-16 16:14:30 +01:00
Starbeamrainbowlabs 1103ae5561
ai: tweak progress bar 2022-09-16 16:07:16 +01:00
Starbeamrainbowlabs 1e35802d2b
ai: fix embed i/o 2022-09-16 16:02:27 +01:00
Starbeamrainbowlabs ed94da7492
fixup 2022-09-16 15:51:26 +01:00
Starbeamrainbowlabs 366db658a8
ds predict: fix filenames in 2022-09-16 15:45:22 +01:00
Starbeamrainbowlabs e333dcba9c
tweak projection head 2022-09-16 15:36:01 +01:00
Starbeamrainbowlabs 6defd24000
bugfix: too many values to unpack 2022-09-15 19:56:17 +01:00
Starbeamrainbowlabs e3c8277255
ai: tweak the segmentation model structure 2022-09-15 19:54:50 +01:00
Starbeamrainbowlabs 1bc8a5bf13
ai: fix crash 2022-09-15 19:37:06 +01:00
Starbeamrainbowlabs bd64986332
ai: implement batched_iterator to replace .batch()
...apparently .batch() means you get a BatchedDataset or whatever when you iterate it like a tf.function instead of the actual tensor :-/
2022-09-15 19:16:38 +01:00
Starbeamrainbowlabs ccd256c00a
embed rainfall radar, not both 2022-09-15 17:37:04 +01:00
Starbeamrainbowlabs 2c74676902
predict → predict_on_batch 2022-09-15 17:31:50 +01:00
Starbeamrainbowlabs f036e79098
fixup 2022-09-15 17:09:26 +01:00
Starbeamrainbowlabs d5f1a26ba3
disable prefetching when predicting a thing 2022-09-15 17:09:09 +01:00
Starbeamrainbowlabs 8770638022
ai: call .predict(), not the model itself 2022-09-14 17:41:01 +01:00
Starbeamrainbowlabs a96cefde62
ai: predict oops 2022-09-14 17:37:48 +01:00
Starbeamrainbowlabs fa3165a5b2
dataset: simplify dataset_predict 2022-09-14 17:33:17 +01:00
Starbeamrainbowlabs 279e27c898
fixup 2022-09-14 17:16:49 +01:00
Starbeamrainbowlabs fad3313ede
fixup 2022-09-14 17:14:04 +01:00
Starbeamrainbowlabs 1e682661db
ai: kwargs in from_checkpoint 2022-09-14 17:11:06 +01:00
Starbeamrainbowlabs 6bda24d4da
ai: how did I miss that?!
bugfix ah/c
2022-09-14 16:53:43 +01:00
Starbeamrainbowlabs decdd434d8
ai from_checkpoint: bugfix 2022-09-14 16:49:01 +01:00
Starbeamrainbowlabs c9e00ea485
pretrain_predict: fix import 2022-09-14 16:02:36 +01:00
Starbeamrainbowlabs f568d8d19f
io.open → handle_open
this was we get transparent .gz support
2022-09-14 16:01:22 +01:00
Starbeamrainbowlabs 9b25186541
fix --only-gpu 2022-09-14 15:55:21 +01:00
Starbeamrainbowlabs a9c9c70d13
typo 2022-09-14 15:17:59 +01:00
Starbeamrainbowlabs fb8f884487
add umap dependencies 2022-09-14 15:16:45 +01:00
Starbeamrainbowlabs 1876a8883c
ai pretrain-predict: fix - → _ in cli parsing 2022-09-14 15:12:07 +01:00
Starbeamrainbowlabs f97b771922
make --help display help 2022-09-14 15:03:07 +01:00
Starbeamrainbowlabs 3c1aef5913
update help 2022-09-13 19:36:56 +01:00
Starbeamrainbowlabs 206257f9f5
ai pretrain_predict: no need to plot here anymore 2022-09-13 19:35:44 +01:00
Starbeamrainbowlabs 7685ec3e8b
implement ability to embed & plot pretrained embeddings 2022-09-13 19:18:59 +01:00
Starbeamrainbowlabs 7130c4fdf8
start implementing core image segmentation model 2022-09-07 17:45:38 +01:00
Starbeamrainbowlabs 22620a1854
ai: implement saving only the rainfall encoder 2022-09-06 19:48:46 +01:00
Starbeamrainbowlabs 4c4358c3e5
whitespace 2022-09-06 16:24:11 +01:00
Starbeamrainbowlabs 4202821d98
typo 2022-09-06 15:37:36 +01:00
Starbeamrainbowlabs 3e13ad12c8
ai Bugfix LayerContrastiveEncoder: channels → input_channels
for consistency
2022-09-05 23:53:16 +01:00
Starbeamrainbowlabs ead8009425
pretrain: add CLI arg for size of watch prediction width/height 2022-09-05 15:36:40 +01:00
Starbeamrainbowlabs 9d39215dd5
dataset: drop incomplete batches 2022-09-05 15:36:10 +01:00
Starbeamrainbowlabs c94f5d042e
ai: slurm fixup again 2022-09-02 19:13:56 +01:00
Starbeamrainbowlabs 7917820e59
ai: slurm fixup 2022-09-02 19:11:54 +01:00
Starbeamrainbowlabs cd104190e8
slurm: no need to be exclusive anymore 2022-09-02 19:09:45 +01:00
Starbeamrainbowlabs a9dede7bfe
ai: n=10 slurm 2022-09-02 19:09:07 +01:00
Starbeamrainbowlabs 42de502f99
slurm-pretrain: set name 2022-09-02 19:08:16 +01:00
Starbeamrainbowlabs 457c2eef0d
ai: fixup 2022-09-02 19:06:54 +01:00
Starbeamrainbowlabs a3f03b6d8d
slurm-pretrain: prepare 2022-09-02 19:05:18 +01:00
Starbeamrainbowlabs 1d1533d160
ai: how did things get this confusing 2022-09-02 18:51:46 +01:00
Starbeamrainbowlabs 1c5defdcd6
ai: cats 2022-09-02 18:45:23 +01:00
Starbeamrainbowlabs 6135bcd0cd
fixup 2022-09-02 18:41:31 +01:00