Skip to content

Commit dbd6364

Browse files
authored
Update config.py
1 parent a599652 commit dbd6364

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,7 @@ class TrainConfig(BaseConfig):
6767
manipulate_seed: int = 0
6868
accum_batches: int = 1
6969
autoenc_mid_attn: bool = True
70-
batch_size: int = 16
70+
batch_size: int = 1
7171
batch_size_eval: int = None
7272
beatgans_gen_type: GenerativeType = GenerativeType.ddim
7373
beatgans_loss_type: LossType = LossType.mse

0 commit comments

Comments
 (0)