Skip to content

Commit

Permalink
Bug: network pkl shouldn't necessarily exist, so remove type
Browse files Browse the repository at this point in the history
  • Loading branch information
PDillis committed Apr 11, 2021
1 parent f921625 commit 5037e59
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 5 deletions.
4 changes: 2 additions & 2 deletions generate.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@ def main():

@main.command(name='images')
@click.pass_context
@click.option('--network', 'network_pkl', type=click.Path(exists=True, dir_okay=False), help='Network pickle filename', required=True)
@click.option('--network', 'network_pkl', help='Network pickle filename', required=True)
@click.option('--seeds', type=num_range, help='List of random seeds')
@click.option('--trunc', 'truncation_psi', type=float, help='Truncation psi', default=1, show_default=True)
@click.option('--class', 'class_idx', type=int, help='Class label (unconditional if not specified)')
Expand Down Expand Up @@ -162,7 +162,7 @@ def _parse_slowdown(slowdown: Union[str, int]) -> int:

@main.command(name='random-video')
@click.pass_context
@click.option('--network', 'network_pkl', type=click.Path(exists=True, dir_okay=False), help='Network pickle filename', required=True)
@click.option('--network', 'network_pkl', help='Network pickle filename', required=True)
@click.option('--seeds', type=num_range, help='List of random seeds', required=True)
@click.option('--trunc', 'truncation_psi', type=float, help='Truncation psi', default=1, show_default=True)
@click.option('--class', 'class_idx', type=int, help='Class label (unconditional if not specified)')
Expand Down
2 changes: 1 addition & 1 deletion sightseeding.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ def _parse_seeds(s: str) -> List[int]:

@click.command()
@click.pass_context
@click.option('--network', '-net', 'network_pkl', type=click.Path(exists=True), help='Network pickle filename', required=True)
@click.option('--network', '-net', 'network_pkl', help='Network pickle filename', required=True)
@click.option('--seeds', '-s', type=_parse_seeds, help='List of seeds to visit in order ("a,b,c", "a-b", "a,b-c,d,e-f,a", ...', required=True)
@click.option('--class', 'class_idx', type=int, help='Class label (unconditional if not specified)')
@click.option('--trunc', 'truncation_psi', type=float, help='Truncation psi', default=1, show_default=True)
Expand Down
4 changes: 2 additions & 2 deletions style_mixing.py
Original file line number Diff line number Diff line change
Expand Up @@ -87,7 +87,7 @@ def main():


@main.command(name='grid')
@click.option('--network', 'network_pkl', type=click.Path(exists=True, dir_okay=False), help='Network pickle filename', required=True)
@click.option('--network', 'network_pkl', help='Network pickle filename', required=True)
@click.option('--row-seeds', '-rows', 'row_seeds', type=num_range, help='Random seeds to use for image rows', required=True)
@click.option('--col-seeds', '-cols', 'col_seeds', type=num_range, help='Random seeds to use for image columns', required=True)
@click.option('--styles', 'col_styles', type=num_range, help='Style layers to use; can pass "coarse", "middle", "fine", or a list or range of ints', default='0-6', show_default=True)
Expand Down Expand Up @@ -175,7 +175,7 @@ def generate_style_mix(

@main.command(name='video')
@click.pass_context
@click.option('--network', 'network_pkl', type=click.Path(exists=True, dir_okay=False),help='Network pickle filename', required=True)
@click.option('--network', 'network_pkl', help='Network pickle filename', required=True)
@click.option('--row-seed', '-row', 'row_seed', type=int, help='Random seed to use for video row', required=True)
@click.option('--col-seeds', '-cols', 'col_seeds', type=num_range, help='Random seeds to use for image columns', required=True)
@click.option('--styles', 'col_styles', type=num_range, help='Style layers to use; can pass "coarse", "middle", "fine", or a list or range of ints', default='0-6', show_default=True)
Expand Down

0 comments on commit 5037e59

Please sign in to comment.