diff options
| -rw-r--r-- | train_dreambooth.py | 2 | ||||
| -rw-r--r-- | train_ti.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index f26b7f5..df8b54c 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
| @@ -110,7 +110,7 @@ def parse_args(): | |||
| 110 | parser.add_argument( | 110 | parser.add_argument( |
| 111 | "--vector_dropout", | 111 | "--vector_dropout", |
| 112 | type=int, | 112 | type=int, |
| 113 | default=0.1, | 113 | default=0, |
| 114 | help="Vector dropout probability.", | 114 | help="Vector dropout probability.", |
| 115 | ) | 115 | ) |
| 116 | parser.add_argument( | 116 | parser.add_argument( |
diff --git a/train_ti.py b/train_ti.py index cacbbc7..1685dc4 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -157,7 +157,7 @@ def parse_args(): | |||
| 157 | parser.add_argument( | 157 | parser.add_argument( |
| 158 | "--vector_dropout", | 158 | "--vector_dropout", |
| 159 | type=int, | 159 | type=int, |
| 160 | default=0.1, | 160 | default=0, |
| 161 | help="Vector dropout probability.", | 161 | help="Vector dropout probability.", |
| 162 | ) | 162 | ) |
| 163 | parser.add_argument( | 163 | parser.add_argument( |