16 Commits

Author	SHA1	Message	Date
Xinlei Chen	4b4d64acce	add options for predictor bn; add checkpoint reloading.	2021-07-12 23:28:07 -07:00
Xinlei Chen	c77dab10a6	xavier init for conv1	2021-07-12 00:20:01 -07:00
Xinlei Chen	84c7fc0fed	add note, and temporarily tensorboard	2021-07-09 15:34:23 -07:00
Xinlei Chen	519aa28fb8	change loss design	2021-07-09 13:32:59 -07:00
Xinlei Chen	0c5d00a3d5	change loss design	2021-07-09 12:41:16 -07:00
Xinlei Chen	e138d15dac	change loss design	2021-07-09 12:28:53 -07:00
Xinlei Chen	da03793c42	define mlp and simplify the code	2021-07-09 11:47:28 -07:00
Xinlei Chen	1d76a72e35	start readme	2021-07-08 04:34:54 -07:00
Xinlei Chen	fe5f5a2395	only the first GPU saves checkpoint	2021-07-05 02:19:09 -07:00
Xinlei Chen	7fbc15248a	reinitialize mlp	2021-07-01 12:54:43 -07:00
Xinlei Chen	f9e4464bf6	add a bunch of things	2021-06-27 03:22:50 -07:00
Xinlei Chen	ae95fdc282	in-progress: add vit	2021-06-24 03:44:50 -07:00
Xinlei Chen	0ec146f8eb	add momentum schedule	2021-06-23 23:26:16 -07:00
Xinlei Chen	efbc756baa	get it running	2021-06-17 02:39:28 -07:00
Xinlei Chen	dacd917eb7	add moco v3	2021-06-17 02:09:43 -07:00
Xinlei Chen	0b658b95cf	add initial files	2021-06-16 19:59:59 -07:00