Skip to content

Commit e4dec08

Browse files
author
Michael Fuest
committed
test updates
1 parent aa3725a commit e4dec08

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

tests/test_configs/model/acgan.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,5 @@ noise_dim: 256
44
cond_emb_dim: 16
55
include_auxiliary_losses: True
66
context_reconstruction_loss_weight: 0.1
7+
tc_loss_weight: 0.0
78
sampling_batch_size: 1

tests/test_configs/model/diffusion_ts.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
_target_: generator.diffusion_ts.gaussian_diffusion.Diffusion_TS
22
name: diffusion_ts
33
context_reconstruction_loss_weight: 0.1
4+
tc_loss_weight: 0.0
45
noise_dim: 256
56
cond_emb_dim: 16
67
n_layer_enc: 4

0 commit comments

Comments
 (0)