16 Commits

Author	SHA1	Message	Date
James Betker	d186414566	More spring cleaning	2022-03-16 12:04:00 -06:00
James Betker	4914c526dc	More cleanup	2021-09-29 14:24:49 -06:00
James Betker	f86df53ce0	Export extract_byol_model as a function	2021-08-05 22:15:26 -06:00
James Betker	fb405d9ef1	CIFAR stuff ... - Extract coarse labels for the CIFAR dataset - Add simple resnet that branches lower layers based on coarse labels - Some other cleanup	2021-06-05 14:16:02 -06:00
James Betker	f129eaa39e	Clean up byol a bit ... - Remove option to aug in dataset (there's really no reason for this now that kornia works on GPU on windows) - Other stufff	2021-05-24 21:35:46 -06:00
James Betker	119f17c808	Add testing capabilities for segformer & contrastive feature	2021-04-27 09:59:50 -06:00
James Betker	17555e7d07	misc adjustments for stylegan	2021-04-21 18:14:17 -06:00
James Betker	b687ef4cd0	Misc	2021-04-21 18:09:46 -06:00
James Betker	94e069bced	Misc changes	2021-03-13 10:45:26 -07:00
James Betker	dac7d768fa	test uresnet playground mods	2021-01-23 13:46:43 -07:00
James Betker	557cdec116	misc	2021-01-23 13:45:17 -07:00
James Betker	34f8c8641f	Support training imagenet classifier	2021-01-11 20:09:16 -07:00
James Betker	14a868e8e6	byol playground updates	2021-01-09 20:54:21 -07:00
James Betker	5a8156026a	Did anyone ask for k-means clustering? ... This is so cool...	2021-01-07 22:37:41 -07:00
James Betker	659814c20f	BYOL script updates	2021-01-07 16:31:28 -07:00
James Betker	9680294430	Move byol scripts around	2021-01-06 14:52:17 -07:00