Created
November 15, 2019 16:14
-
-
Save williamFalcon/b350d4e11d512475143db227176b1132 to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| train_arg_parser.add_argument("--epochs", type=int, default=2, | |
| help="number of training epochs, default is 2") | |
| train_arg_parser.add_argument("--batch-size", type=int, default=4, | |
| help="batch size for training, default is 4") | |
| train_arg_parser.add_argument("--dataset", type=str, required=True, | |
| help="path to training dataset, the path should point to a folder " | |
| "containing another folder with all the training images") | |
| train_arg_parser.add_argument("--style-image", type=str, default="images/style-images/mosaic.jpg", | |
| help="path to style-image") | |
| train_arg_parser.add_argument("--image-size", type=int, default=256, | |
| help="size of training images, default is 256 X 256") | |
| train_arg_parser.add_argument("--style-size", type=int, default=None, | |
| help="size of style-image, default is the original size of style image") | |
| train_arg_parser.add_argument("--content-weight", type=float, default=1e5, | |
| help="weight for content-loss, default is 1e5") | |
| train_arg_parser.add_argument("--style-weight", type=float, default=1e10, | |
| help="weight for style-loss, default is 1e10") | |
| train_arg_parser.add_argument("--lr", type=float, default=1e-3, | |
| help="learning rate, default is 1e-3") |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment