We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a599652 commit dbd6364Copy full SHA for dbd6364
config.py
@@ -67,7 +67,7 @@ class TrainConfig(BaseConfig):
67
manipulate_seed: int = 0
68
accum_batches: int = 1
69
autoenc_mid_attn: bool = True
70
- batch_size: int = 16
+ batch_size: int = 1
71
batch_size_eval: int = None
72
beatgans_gen_type: GenerativeType = GenerativeType.ddim
73
beatgans_loss_type: LossType = LossType.mse
0 commit comments