47 Commits

Author	SHA1	Message	Date
flytocc	8b8a02d607	add update_freq option for gradient accumulation	2022-04-28 00:50:28 +08:00
flytocc	ed8202236b	add EMA code	2022-04-28 00:39:53 +08:00
gaotingquan	b86ddf470e	fix: when eval only, AMP can be enabled	2022-04-22 17:49:44 +08:00
HydrogenSulfate	aa26a8c1d8	fix bug for static graph	2022-04-21 16:31:28 +08:00
HydrogenSulfate	3a1276d315	train_loss_func only used in train mode	2022-04-19 19:54:48 +08:00
HydrogenSulfate	24abea151a	support for multi optimizer case	2022-04-19 14:26:42 +08:00
dongshuilong	a944603da0	fix log twice bug	2022-03-30 08:31:35 +00:00
huangqipeng	b62b98d79f	feat: support mlu device and amp of mlu	2022-03-14 15:48:26 +08:00
littletomatodonkey	f68c098a4a	fix train acc log	2022-03-09 19:58:36 +08:00
WangChen0902	7595ba6d70	add AFD (#1683) ... * add AFD	2022-02-28 19:11:50 +08:00
Tingquan Gao	42134cd8dd	fix: raise warning when using Global.class_num	2022-01-25 15:06:36 +08:00
Tingquan Gao	bb6581d21b	refactor: raise warning when gpu numbers is not 4	2022-01-25 15:06:36 +08:00
Tingquan Gao	8f0bd5b582	fix: fix vdl makedir	2022-01-25 15:06:36 +08:00
gaotingquan	10c93c55d1	fix: enable amp only in training	2022-01-25 11:58:07 +08:00
gaotingquan	7040ce8314	refactor: change params to be consistent with amp	2022-01-25 11:58:07 +08:00
zhangbo9674	cd039a7b37	add save_dtype	2022-01-10 18:19:03 +08:00
zhangbo9674	b2956c1b41	refine code	2022-01-10 18:19:03 +08:00
zhangbo9674	205592a3e3	fix amp with distribute bug	2022-01-10 18:19:03 +08:00
littletomatodonkey	aea712cc87	add dist of rec model (#1574) ... * add distillation loss func and rec distillation	2022-01-05 19:25:36 +08:00
gaotingquan	6e13ff3068	fix: use hasattr() to check if collate_fn is in dataloader ... fix bug caused by PR #1596	2021-12-30 16:35:05 +08:00
gaotingquan	7da2a997e9	fix: save latest model every epoch	2021-12-27 22:04:26 +08:00
gaotingquan	5d53e9f152	fix: raise warning when setting batch_transform_ops and TopkAcc	2021-12-24 21:22:52 +08:00
zhangbo9674	b54ee04491	Accelerate dynamic graph amp training	2021-12-20 06:36:56 +00:00
weishengyu	534037d145	dbg	2021-12-10 11:14:14 +08:00
weishengyu	ad1a2fd137	move slim into arch	2021-12-09 20:08:57 +08:00
weishengyu	6c5d1ebc28	add pruner and quanter for theseus	2021-12-09 14:51:40 +08:00
cuicheng01	a3b54f15d5	fix export quant_model	2021-12-03 07:39:31 +00:00
stephon	7a17f72fc2	fix seed=0 bug	2021-11-01 06:16:24 +00:00
gaotingquan	ed459a2a16	refactor: adapt to static graph in deprecating MixCELoss	2021-10-27 19:47:43 +08:00
gaotingquan	ba2dd01a13	refactor: deprecate MixCELoss	2021-10-27 19:47:43 +08:00
ronnywang	a0eb34a642	Add npu supporting (#1324)	2021-10-22 11:02:29 +08:00
dongshuilong	06382519cf	fix train without eval bug	2021-10-21 02:39:27 +00:00
dongshuilong	0e20e60f6b	fix slim load pretrained model bug	2021-10-18 10:07:14 +00:00
cuicheng01	af9aae730e	add multilabel feature	2021-09-26 07:05:13 +00:00
cuicheng01	ba9b708acb	Merge pull request #1174 from TingquanGao/dev/add_adamw ... feat: add AdamW	2021-09-17 18:53:18 +08:00
Wei Shengyu	d98b881603	Merge pull request #1208 from weisy11/fix_bug_of_theseus ... dbg theseus	2021-09-15 14:04:51 +08:00
weishengyu	600067f4fc	dbg	2021-09-15 11:35:49 +08:00
dongshuilong	94c9900ece	add seed	2021-09-14 04:06:37 +00:00
Walter	74622af482	Merge pull request #1093 from RainFrost1/slim ... 添加slim功能	2021-09-13 19:11:43 +08:00
Walter	ce39aea97f	Merge pull request #1187 from RainFrost1/retrieval_dataloader ... fix issues when gallery == query dataset	2021-09-03 14:33:23 +08:00
dongshuilong	50f2547005	Merge branch 'develop' of https://github.com/PaddlePaddle/PaddleClas into slim	2021-09-03 03:19:14 +00:00
dongshuilong	d49657ad08	fix rec forward bug	2021-09-02 07:42:22 +00:00
gaotingquan	079434dc5f	feat: add AdamW	2021-09-01 08:07:48 +00:00
dongshuilong	68c1b089e8	update slim for new trainer	2021-08-27 09:32:37 +00:00
dongshuilong	1a1eb3a1ae	fix issues when gallery == query dataset	2021-08-27 06:25:27 +00:00
dongshuilong	5bf74beb46	update according comments	2021-08-24 07:07:17 +00:00
dongshuilong	15f6f58139	refactor trainer v2	2021-08-24 03:02:55 +00:00