Commit Graph

34 Commits (542b3200086c7163805ebee9f8f85b6c85ce8d8c)

Author SHA1 Message Date
gmm be6e0b7b53
【benchmark】add use_fused_attn=True for benchmark (#3160)
* add max_memory_reserved for benchmark

* change unit

* fix

* fix nvidia-dali-cuda110;DLTP-54994

* add use_fused_attn=True  for benchmark
2024-06-06 11:25:51 +08:00
changdazhou a39351537d
Ctrl mem log (#3107)
* support ctrl print memory info

* support ctrl print memory info

* fix print log bug

* support default print_mem_info=True for tipc
2024-03-12 20:32:27 +08:00
gmm e4abcc08d0
add max_memory_reserved for benchmark (#3042) 2023-11-22 17:31:30 +08:00
duanyanhui 48fc521e61 fix npu tipc scripts to reduce running time 2023-07-06 16:25:16 +08:00
HydrogenSulfate cc638a9d02 update pretrained_model and inference model link in TIPC config and refine TIPC script 2023-02-28 15:01:21 +08:00
gaotingquan d336944566 rm unset CUDA_VISIBLE_DEVICES before training 2022-12-29 11:44:27 +08:00
gaotingquan 983d512a45 tipc: fix the keyword of ShiTu model
In benchmark_train mode, the config name is benchmark.txt. So the keyword of ShiTu should be model_name instead of the config name.
2022-09-19 14:30:54 +08:00
gaotingquan dafcb5cddf tipc: fix log path for GeneralRecognition 2022-09-14 10:44:29 +08:00
qili93 b1c2350a26 [TIPC] add more scripts change, test=develop 2022-09-13 15:03:14 +08:00
gaotingquan 2700beea82 tipc: fix log path 2022-09-08 08:15:33 +00:00
gaotingquan fb669a05b8 tipc: fix log path 2022-09-06 13:25:09 +00:00
gaotingquan b3d26a987d tipc: fix log path 2022-09-05 15:57:49 +00:00
gaotingquan 3d1fb324a3 tipc: support AMP O2 2022-08-05 14:34:14 +08:00
HydrogenSulfate 27dd1bc73c merge develop and solve conflicts 2022-06-27 15:48:36 +08:00
HydrogenSulfate e7d9ba5874 update config and script 2022-06-24 04:25:56 +00:00
HydrogenSulfate e6cea06326 format test_train_inference_python.sh 2022-06-23 07:48:17 +00:00
HydrogenSulfate 4400aa2747 debug 2022-06-23 06:50:46 +00:00
HydrogenSulfate 480e40eb73 add main model KL transformation chain 2022-06-23 06:46:12 +00:00
HydrogenSulfate 2b758dec3f add pact chain configs and refine related scripts 2022-06-22 17:45:58 +00:00
HydrogenSulfate e0a4307a44 update status_check function 2022-06-09 16:22:08 +08:00
HydrogenSulfate e8ced77fdb add train_fleet_infer chain 2022-06-07 17:49:13 +08:00
Aurelius84 33a639dd3b [TIPC-Benchmark]Support @to_static traing for Benchmark 2022-03-14 08:18:19 +00:00
Walter 498717492b
Merge pull request #1726 from RainFrost1/tipc_save_path
add model_name fot tipc save_path
2022-03-04 14:04:03 +08:00
dongshuilong ccea789eef add model_name fot tipc save_path 2022-03-04 03:14:16 +00:00
dongshuilong 18a76f491e remove cudnn_deterministic in tipc 2022-03-04 02:54:04 +00:00
dongshuilong dc6281a6d4 add benchmark for tipc 2022-02-10 08:25:52 +00:00
dongshuilong f6f2a20126 fix tipc lite_train_whole_infer bug 2021-12-24 03:11:26 +00:00
dongshuilong 5035240cd7 merge develop 2021-12-14 06:57:03 +00:00
dongshuilong a143044b7d change large model batch_size for tipc 2021-12-13 12:28:07 +00:00
dongshuilong e28b16f99d fix tipc batch_size bug 2021-12-09 12:29:00 +00:00
dongshuilong a3465a6591 add rec model for tipc 2021-11-25 12:50:13 +00:00
dongshuilong b9440ce5eb update wholechain klquant 2021-11-17 12:43:38 +00:00
dongshuilong 62b3775e92 update tipc and add whole model for train_infer_python 2021-11-17 19:02:03 +08:00
stephon 79bbc286aa add test_tipc 2021-11-10 11:38:22 +00:00