4a5eef64d2a0ab5e1737ffbb345d9e72d2f1adc9,train.py,,,#,56

Before Change

parser = argparse.ArgumentParser()
// Basic options
parser.add_argument("--gpu", type=int, default=-1)
parser.add_argument("--content_dir", type=str, required=True,
                    help="Directory path to a batch of content images")
parser.add_argument("--style_dir", type=str, required=True,
                    help="Directory path to a batch of style images")

After Change

parser.add_argument("--save_model_interval", type=int, default=10000)
args = parser.parse_args()

device = torch.device("cuda")

if not os.path.exists(args.save_dir):
    os.mkdir(args.save_dir)

Instances