diff --git a/ssd/train.py b/ssd/train.py index dad32ecd46938edccba3a7557bc822a02fa8a231..bfcbf9115ffa44cd70f6cd3ed4ed10f05d139c4d 100644 --- a/ssd/train.py +++ b/ssd/train.py @@ -58,7 +58,7 @@ def train(train_file_list, dev_file_list, data_args, init_model_path): sys.stdout.flush() if isinstance(event, paddle.event.EndPass): - with gzip.open('checkpionts/params_pass_%05d.tar.gz' % \ + with gzip.open('checkpoints/params_pass_%05d.tar.gz' % \ event.pass_id, 'w') as f: parameters.to_tar(f) result = trainer.test(reader=dev_reader, feeding=feeding) diff --git a/ssd/vgg_ssd_net.py b/ssd/vgg_ssd_net.py index 0a4f0d314a685b957cb0a1c2f0a74efc77bc477a..cc7be564ce5997baa86ab4673b86aae06cc80d8d 100644 --- a/ssd/vgg_ssd_net.py +++ b/ssd/vgg_ssd_net.py @@ -1,5 +1,5 @@ import paddle.v2 as paddle -from config.vgg_config import cfg +from config.pascal_voc_conf import cfg def net_conf(mode):