style checks

This commit is contained in:
rasbt
2024-04-24 07:48:51 -05:00
parent 46d09b30d9
commit 258aff3e9a

View File

@@ -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
###############################