Skip to content

Commit 8b21aa6

Browse files
author
james
committed
fix greedyopt
1 parent 536a61f commit 8b21aa6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

teal/greedyopt.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,7 @@ def process_layer(layer, model_type, layer_idx, target_sparsity, base_step_size,
101101

102102
histogram_path = os.path.join(teal_path, 'histograms')
103103
activations_path = os.path.join(teal_path, 'activations', f'act_{layer_idx}.pt')
104-
output_path = os.path.join(teal_path, 'lookup', f'layer_{layer_idx}.csv')
104+
output_path = os.path.join(teal_path, 'lookup', f'layer-{layer_idx}', 'results.csv')
105105

106106
device = "cuda"
107107
input_acts = torch.load(activations_path, map_location='cpu').to(device)

0 commit comments

Comments
 (0)