tdrussell

The following test script works for me and successfully runs with 2 GPUs: ``` import torch import deepspeed import argparse def parse_args(): parser = argparse.ArgumentParser() parser = deepspeed.add_config_arguments(parser) parser.add_argument('--local_rank', type=int,...