Commit Graph

9 Commits (85f65ce76f664e5b4fc50f6ee6351a00b7a8dd18)

Author SHA1 Message Date
HydrogenSulfate 4060957669 fix(dali): add shuffle to prob_list in ExternalSource_RandomIdentity 2022-12-13 14:27:15 +08:00
HydrogenSulfate b22752e2ea refactor DALI 2022-12-13 14:27:15 +08:00
HydrogenSulfate a14df4ac52 fix tensor conversion in static mode with dali loader 2022-10-20 12:03:03 +00:00
HydrogenSulfate 61b4153907 add batch Tensor collate to simplify dali code in train/eval/retrival code 2022-10-13 12:05:50 +08:00
gaotingquan 3f8f981423 fix: support specify DALI threads num 2022-09-01 20:21:43 +08:00
dongshuilong 49f010a1c6 add r50 dynamic fp16 train for benchmark 2022-06-15 14:33:17 +00:00
dongshuilong 11b0860a1b fix dali env bug 2022-05-11 03:06:55 +00:00
littletomatodonkey 9d9cd3719e
add static training (#1037)
* add static training

* fix typo

* add se fp16

* rm note

* fix loader

* fix cfg
2021-07-15 10:30:07 +08:00
Walter 73004f78f5
add fp16 amp training and dali (#993)
* add dygraph amp train
2021-07-14 22:59:23 +08:00