polish code

tools/program.py

@@ -215,9 +215,6 @@ def train(config,
                 preds = model(images, data=batch[1:])
             else:
                 preds = model(images)
-            state_dict = model.state_dict()
-            # for key in state_dict:
-            #     print(key)
             loss = loss_class(preds, batch)
             avg_loss = loss['loss']
             avg_loss.backward()
@@ -414,7 +411,6 @@ def preprocess(is_train=False):
             yaml.dump(
                 dict(config), f, default_flow_style=False, sort_keys=False)
         log_file = '{}/train.log'.format(save_model_dir)
-        print("log has save in {}/train.log".format(save_model_dir))
     else:
         log_file = None
     logger = get_logger(name='root', log_file=log_file)