You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+4-4Lines changed: 4 additions & 4 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -151,7 +151,7 @@ The `run.py` command requires the following arguments:
151
151
*`--epochs`: number of epochs to train, `--epochs 0` means running zero-shot inference.
152
152
*`--bpe`: batches per epoch (replaces the length of the dataloader as default value). `--bpe 100 --epochs 10` means that each epoch consists of 100 batches, and overall training is 1000 batches. Set `--bpe null` to use the full length dataloader or comment the `bpe` line in the yaml configs.
153
153
*`--gpus`: number of gpu devices, set to `--gpus null` when running on CPUs, `--gpus [0]` for a single GPU, or otherwise set the number of GPUs for a [distributed setup](#distributed-setup)
154
-
*`--ckpt`: path to the one of the ULTRA checkpoints to use (you can use those provided in the repo ot trained on your own). Use `--ckpt null` to start training from scratch (or run zero-shot inference on a randomly initialized model, it still might surprise you and demonstrate non-zero performance).
154
+
*`--ckpt`: **full**path to the one of the ULTRA checkpoints to use (you can use those provided in the repo ot trained on your own). Use `--ckpt null` to start training from scratch (or run zero-shot inference on a randomly initialized model, it still might surprise you and demonstrate non-zero performance).
155
155
156
156
Zero-shot inference setup is `--epochs 0` with a given checkpoint `ckpt`.
157
157
@@ -161,12 +161,12 @@ Fine-tuning of a checkpoint is when epochs > 0 with a given checkpoint.
161
161
An example command for an inductive dataset to run on a CPU:
@@ -176,7 +176,7 @@ Using the same config files, you only need to specify:
176
176
177
177
*`-c <yaml config>`: use the full path to the yaml config because workdir will be reset after each dataset;
178
178
*`-d, --datasets`: a comma-separated list of [datasets](#datasets) to run, inductive datasets use the `name:version` convention. For example, `-d ILPC2022:small,ILPC2022:large`;
179
-
*`--ckpt`: ULTRA checkpoint to run the experiments on, use the full path to the file;
179
+
*`--ckpt`: ULTRA checkpoint to run the experiments on, use the **full** path to the file;
180
180
*`--gpus`: the same as in [run single](#run-a-single-experiment);
181
181
*`-reps` (optional): number of repeats with different seeds, set by default to 1 for zero-shot inference;
182
182
*`-ft, --finetune` (optional): use the finetuning configs of ULTRA (`default_finetuning_config`) to fine-tune a given checkpoint for specified `epochs` and `bpe`;
0 commit comments