Skip to content

Commit

Permalink
Pre-run for one threshold
Browse files Browse the repository at this point in the history
  • Loading branch information
krishagrawal112 committed Mar 29, 2023
1 parent 4d74fc6 commit a03741e
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
1 change: 1 addition & 0 deletions imagenet_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,6 +142,7 @@ def imagenet_main():
replace_with_vanilla_convolution(model, window_size=args.ma_window_size)
replace_with_variable_relu(model, threshold=args.relu_threshold)
validate(calibrate_loader, model, criterion, args.print_freq)
validate(val_loader, model, criterion)
output_sparsity_to_csv(args.arch, model, args.output_path)

if __name__ == '__main__':
Expand Down
2 changes: 1 addition & 1 deletion run.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@
'''


sweep_range = [0.001, 0.002, 0.005, 0.01, 0.02, 0.05, 0.1, 0.2, 0.5, 1, 2, 5, 10]
sweep_range = [0.001] #, 0.002, 0.005, 0.01, 0.02, 0.05, 0.1, 0.2, 0.5, 1, 2, 5, 10]
for relu_threshold in sweep_range:
test_name = "sparsity_run_relu_threshold" + str(relu_threshold)

Expand Down

0 comments on commit a03741e

Please sign in to comment.