style checks

This commit is contained in:
rasbt 2024-04-24 07:48:51 -05:00
parent d311bae25a
commit 51f4980a42

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