|
|
@@ -254,7 +254,7 @@ if __name__ == "__main__":
|
|
|
help=(
|
|
|
"Whether to use 'pretrained' or 'random' weights."
|
|
|
)
|
|
|
- )
|
|
|
+ )
|
|
|
parser.add_argument(
|
|
|
"--trainable_layers",
|
|
|
type=str,
|
|
|
@@ -290,7 +290,6 @@ if __name__ == "__main__":
|
|
|
else:
|
|
|
raise ValueError("Invalid --trainable_token argument")
|
|
|
|
|
|
-
|
|
|
###############################
|
|
|
# Load model
|
|
|
###############################
|