Skip to content

Commit 19bf57b

Browse files
committed
fix standing_stats typo in train_fns
1 parent eb42d44 commit 19bf57b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

train_fns.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def save_and_sample(G, D, G_ema, z_, y_, fixed_z, fixed_y,
112112

113113
# Accumulate standing statistics?
114114
if config['accumulate_stats']:
115-
utils.accumulate_stats(G_ema if config['ema'] and config['use_ema'] else G,
115+
utils.accumulate_standing_stats(G_ema if config['ema'] and config['use_ema'] else G,
116116
z_, y_, config['n_classes'],
117117
config['num_standing_accumulations'])
118118

@@ -161,7 +161,7 @@ def test(G, D, G_ema, state_dict, config, sample, get_inception_metrics,
161161
experiment_name, test_log):
162162
print('Gathering inception metrics...')
163163
if config['accumulate_stats']:
164-
utils.accumulate_stats(G_ema if config['ema'] and config['use_ema'] else G,
164+
utils.accumulate_standing_stats(G_ema if config['ema'] and config['use_ema'] else G,
165165
z_, y_, config['n_classes'],
166166
config['num_standing_accumulations'])
167167
IS_mean, IS_std, FID = get_inception_metrics(sample,

0 commit comments

Comments
 (0)