Skip to content

Commit

Permalink
Indent
Browse files Browse the repository at this point in the history
  • Loading branch information
glenn-jocher committed Jun 28, 2021
1 parent 7b62f38 commit cd003e9
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions utils/plots.py
Original file line number Diff line number Diff line change
Expand Up @@ -450,10 +450,10 @@ def plot_results(start=0, stop=0, bucket='', id=(), labels=(), save_dir=''):

def feature_visualization(features, module_type, module_idx, n=64):
"""
features: Features to be visualized
module_type: module type
module_idx: module layer index within model
n: Maximum number of feature maps to plot
features: Features to be visualized
module_type: Module type
module_idx: Module layer index within model
n: Maximum number of feature maps to plot
"""
project, name = 'runs/features', 'exp'
save_dir = increment_path(Path(project) / name) # increment run
Expand Down

0 comments on commit cd003e9

Please sign in to comment.