Wiki Task fix

Author: wdykas

tasks/finetune_utils.py

@@ -13,10 +13,10 @@
 from megatron.core.enums import ModelType
 from megatron.training.checkpointing import load_checkpoint
 from megatron.training.checkpointing import save_checkpoint
-from megatron.training import evaluate_and_print_results
-from megatron.training import setup_model_and_optimizer
-from megatron.training import train_step
-from megatron.training import training_log
+from megatron.training.training import evaluate_and_print_results
+from megatron.training.training import setup_model_and_optimizer
+from megatron.training.training import train_step
+from megatron.training.training import training_log
 from megatron.training.utils import average_losses_across_data_parallel_group
 from megatron.training.utils import calc_params_l2_norm
 from megatron.training.utils import check_adlr_autoresume_termination

tasks/main.py

@@ -20,8 +20,6 @@ def get_tasks_args(parser):
     group.add_argument('--epochs', type=int, default=None,
                        help='Number of finetunning epochs. Zero results in '
                        'evaluation only.')
-    group.add_argument('--pretrained-checkpoint', type=str, default=None,
-                       help='Pretrained checkpoint used for finetunning.')
     group.add_argument('--keep-last', action='store_true',
                        help='Keep the last batch (maybe incomplete) in'
                        'the data loader')