You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
ougongchang 33b5cda1da Decide whether to collect data by dataset sink mode and current step in SummaryCollector. 5 years ago
..
callback Decide whether to collect data by dataset sink mode and current step in SummaryCollector. 5 years ago
quant fix bug in quant deploy export 5 years ago
summary change some comment name in the whole project 5 years ago
__init__.py initial version 5 years ago
_utils.py Add a callback named SummaryCollector and delete SummaryStep callback 5 years ago
amp.py fix the problem of BatchNorm config failure at Amp O3 level and some unexpected indent 5 years ago
dataset_helper.py fix loop size error when exec train after model init 5 years ago
loss_scale_manager.py Add cell name to error message 5 years ago
model.py !2326 fix the problem of BatchNorm config failure at Amp O3 level and some unexpected indent 5 years ago
parallel_utils.py initial version 5 years ago
serialization.py fix checkpoint evaliaction. 5 years ago