We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7bb5961 commit e0d4981Copy full SHA for e0d4981
1 file changed
training/train.py
@@ -16,7 +16,7 @@
16
dataset = download.download_deep_scores()
17
train.train_segnet()
18
elif model_type == "transformer":
19
- transformer.train_transformer(resume="checkpoint-24552")
+ transformer.train_transformer()
20
else:
21
eprint("Unknown model: " + model_type)
22
sys.exit(1)
0 commit comments