105 commits

Author	SHA1	Message	Date
Starbeamrainbowlabs	cc6679c609	batch data; use generator	2022-10-20 15:22:29 +01:00
Starbeamrainbowlabs	d306853c42	use right daataset	2022-10-20 15:16:24 +01:00
Starbeamrainbowlabs	200076596b	finish train_predict	2022-10-19 17:26:40 +01:00
Starbeamrainbowlabs	488f78fca5	pretrain_predict: default to parallel_reads=0	2022-10-19 16:59:45 +01:00
Starbeamrainbowlabs	fe43ddfbf9	start implementing driver for train_predict, but not finished yet	2022-10-18 19:37:55 +01:00
Starbeamrainbowlabs	f12e6ab905	No need for a CLI arg for feature_dim_in - metadata should contain this	2022-10-13 17:37:16 +01:00
Starbeamrainbowlabs	32f5200d3b	pass model_arch properly	2022-10-12 16:50:06 +01:00
Starbeamrainbowlabs	5933fb1061	fixup	2022-10-11 19:23:41 +01:00
Starbeamrainbowlabs	f4a2c742d9	typo	2022-10-11 19:19:23 +01:00
Starbeamrainbowlabs	11f91a7cf4	train: add --arch; default to convnext_i_xtiny	2022-10-11 19:18:01 +01:00
Starbeamrainbowlabs	5666c5a0d9	typo	2022-10-10 18:12:51 +01:00
Starbeamrainbowlabs	131c0a0a5b	pretrain-predict: create dir if not exists	2022-10-10 18:00:55 +01:00
Starbeamrainbowlabs	f883986eaa	Bugfix: modeset to enable TFRecordWriter instead of bare handle	2022-10-06 20:07:59 +01:00
Starbeamrainbowlabs	9f3ae96894	finish wiring for --water-size	2022-10-06 19:21:50 +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	d765b3b14e	fix crash	2022-09-27 18:43:43 +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	4fd852d782	fixup	2022-09-16 16:44:35 +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	1bc8a5bf13	ai: fix crash	2022-09-15 19:37:06 +01:00
Starbeamrainbowlabs	a96cefde62	ai: predict oops	2022-09-14 17:37:48 +01:00
Starbeamrainbowlabs	1e682661db	ai: kwargs in from_checkpoint	2022-09-14 17:11:06 +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	1876a8883c	ai pretrain-predict: fix - → _ in cli parsing	2022-09-14 15:12:07 +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	4c4358c3e5	whitespace	2022-09-06 16:24:11 +01:00
Starbeamrainbowlabs	4202821d98	typo	2022-09-06 15:37:36 +01:00
Starbeamrainbowlabs	ead8009425	pretrain: add CLI arg for size of watch prediction width/height	2022-09-05 15:36:40 +01:00
Starbeamrainbowlabs	c33c0a0899	ai: these shapes are so annoying	2022-09-02 18:39:24 +01:00
Starbeamrainbowlabs	88acd54a97	ai: off-by-one	2022-09-02 18:13:48 +01:00
Starbeamrainbowlabs	23bcd294c4	AI: 0.75 → 0.5?	2022-09-02 18:12:51 +01:00
Starbeamrainbowlabs	b0e1aeac35	ai: knew it	2022-09-02 18:10:29 +01:00
Starbeamrainbowlabs	ef0f15960d	send correct water shape to	2022-09-02 18:09:36 +01:00
Starbeamrainbowlabs	c78b6562ef	debug adjustment	2022-09-02 17:00:21 +01:00
Starbeamrainbowlabs	51cf08a386	ResNetRSV2 → ConvNeXt ... ironically this makes the model simpler o/	2022-08-31 18:51:01 +01:00
Starbeamrainbowlabs	dbf929325a	typo; add pretrain slurm job file	2022-08-31 16:32:17 +01:00
Starbeamrainbowlabs	15a3519107	ai: the best thing about implementing a model is that you don't have to test it on the same day :P	2022-08-11 18:26:28 +01:00