Skip to content

Commit

Permalink
Merge pull request #68 from omidforoqi/patch-1
Browse files Browse the repository at this point in the history
fix worker numbers type
  • Loading branch information
laurahanu authored Oct 19, 2022
2 parents 58f1f80 + 3582971 commit 1f1e8a9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion train.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,7 @@ def cli_main():
parser.add_argument(
"--num_workers",
default=10,
type=str,
type=int,
help="number of workers used in the data loader (default: 10)",
)
parser.add_argument("-e", "--n_epochs", default=100, type=int, help="if given, override the num")
Expand Down

0 comments on commit 1f1e8a9

Please sign in to comment.