Skip to content

Commit

Permalink
Merge pull request #10 from cmubig/cofrida-coco
Browse files Browse the repository at this point in the history
Dataset switched to COCO
  • Loading branch information
pschaldenbrand authored Jan 29, 2024
2 parents 98c59c6 + 5d01696 commit 1bf0205
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/options.py
Original file line number Diff line number Diff line change
Expand Up @@ -85,7 +85,7 @@ def initialize(self, parser):

# CoFRIDA Training Parameters
parser.add_argument("--cofrida_dataset", type=str,
default="laion/laion-art", help='A dataset for training CoFRIDA')
default="ChristophSchuhmann/MS_COCO_2017_URL_TEXT", help='A dataset for training CoFRIDA')
parser.add_argument("--cofrida_background_image", type=str,
default='./blank_canvas.jpg', help='path to image to use as background for cofrida drawings/paintings')
parser.add_argument("--output_parent_dir", type=str,
Expand Down

0 comments on commit 1bf0205

Please sign in to comment.