Skip to content

Commit

Permalink
fix doc
Browse files Browse the repository at this point in the history
  • Loading branch information
littletomatodonkey committed Mar 28, 2022
1 parent 0c89ae2 commit 8ab73a8
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion deploy/slim/quantization/quant.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ def main(config, device, logger, vdl_writer):
model = build_model(config['Architecture'])

pre_best_model_dict = dict()
# load pretrain model
# load fp32 model to begin quantization
if config["Global"]["pretrained_model"] is not None:
pre_best_model_dict = load_model(config, model)

Expand All @@ -139,6 +139,7 @@ def main(config, device, logger, vdl_writer):
step_each_epoch=len(train_dataloader),
parameters=model.parameters())

# resume PACT training process
if config["Global"]["checkpoints"] is not None:
pre_best_model_dict = load_model(config, model, optimizer)

Expand Down

0 comments on commit 8ab73a8

Please sign in to comment.