13 Commits

Author	SHA1	Message	Date
Jiarui Fang	845856ea29	[Graph] building computing graph with ColoTensor, Linear only (#917)	2022-05-07 17:10:37 +08:00
Ziyue Jiang	75d221918a	[Tensor] add 1d vocab loss (#918) ... * add 1d vocab loss * polish	2022-05-07 15:49:14 +08:00
Ziyue Jiang	dfaff4e243	[Tensor] fix test_model (#916) ... * polish test_model * polish	2022-05-06 18:06:22 +08:00
Jiarui Fang	ed6426c300	[Tensor] polish model test (#915)	2022-05-06 17:07:56 +08:00
Ziyue Jiang	0fab86b12a	[Tensor] add a basic bert. (#911) ... * add base bert test * Add bert test * polish * remove test_bert * polish	2022-05-06 15:03:43 +08:00
Jiarui Fang	ab95ec9aea	[Tensor] init ColoParameter (#914)	2022-05-06 12:57:14 +08:00
Ziyue Jiang	f593a5637e	[Tensor] add embedding tp1d row (#904)	2022-04-29 14:10:05 +08:00
Ziyue Jiang	2c0d19d755	[Tensor] add ColoTensor TP1Dcol Embedding (#899)	2022-04-28 17:45:06 +08:00
Jiarui Fang	d16671da75	[Tensor] initialize the ColoOptimizer (#898) ... * [Tensor] activation is an attr of ColoTensor * [Tensor] add optimizer * only detach parameters in context * polish code	2022-04-28 15:23:40 +08:00
Jiarui Fang	e76f76c08b	[Tensor] test parameters() as member function (#896)	2022-04-28 10:57:14 +08:00
Ziyue Jiang	cb182da7c5	[tensor] refine linear and add gather for laynorm (#893) ... * refine linear and add function to ColoTensor * add gather for layernorm * polish * polish	2022-04-28 10:55:40 +08:00
Jiarui Fang	26c49639d8	[Tensor] overriding paramters() for Module using ColoTensor (#889)	2022-04-27 15:28:59 +08:00
Jiarui Fang	a0e5971692	[Tensor] test model check results for a simple net (#887)	2022-04-27 12:00:18 +08:00