LittletreeZou
commited on
Commit
·
f331484
1
Parent(s):
3e376df
initial upload
Browse files- config.yaml +1 -9
config.yaml
CHANGED
|
@@ -2,14 +2,6 @@
|
|
| 2 |
seed_everything: 42
|
| 3 |
trainer:
|
| 4 |
accelerator: auto
|
| 5 |
-
# strategy:
|
| 6 |
-
# class_path: lightning.pytorch.strategies.ModelParallelStrategy
|
| 7 |
-
# init_args:
|
| 8 |
-
# data_parallel_size: 1
|
| 9 |
-
# tensor_parallel_size: 8
|
| 10 |
-
# save_distributed_checkpoint: true
|
| 11 |
-
# process_group_backend: null
|
| 12 |
-
# timeout: 0:30:00
|
| 13 |
strategy:
|
| 14 |
class_path: lightning.pytorch.strategies.FSDPStrategy
|
| 15 |
init_args:
|
|
@@ -201,7 +193,7 @@ data:
|
|
| 201 |
class_path: genbio_finetune.data.ContactPredictionBinary
|
| 202 |
init_args:
|
| 203 |
path: biomap-research/contact_prediction_binary
|
| 204 |
-
batch_size: 1
|
| 205 |
max_length: 512
|
| 206 |
train_split_name: train
|
| 207 |
test_split_name: test
|
|
|
|
| 2 |
seed_everything: 42
|
| 3 |
trainer:
|
| 4 |
accelerator: auto
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
strategy:
|
| 6 |
class_path: lightning.pytorch.strategies.FSDPStrategy
|
| 7 |
init_args:
|
|
|
|
| 193 |
class_path: genbio_finetune.data.ContactPredictionBinary
|
| 194 |
init_args:
|
| 195 |
path: biomap-research/contact_prediction_binary
|
| 196 |
+
batch_size: 1 # only support micro_batch_size=1
|
| 197 |
max_length: 512
|
| 198 |
train_split_name: train
|
| 199 |
test_split_name: test
|