Skip to content

Commit

Permalink
Merge pull request #17 from LukasHedegaard/develop
Browse files Browse the repository at this point in the history
Fix requirements
  • Loading branch information
LukasHedegaard committed Jun 3, 2022
2 parents e1dcdf9 + 71488b9 commit 8ea3dc1
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 3 deletions.
1 change: 0 additions & 1 deletion docs/source/_templates/theme_variables.jinja
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
{%- set external_urls = {
'github': 'https://github.com/LukasHedegaard/ride',
'github_issues': 'https://github.com/LukasHedegaard/ride/issues',
'contributing': 'https://github.com/PyTorchLightning/ride/blob/master/CONTRIBUTING.md',
'docs': 'https://ride.rtfd.io/en/latest',
'home': 'https://ride.rtfd.io/en/latest/',
'get_started': 'https://ride.readthedocs.io/en/latest/starter/readme.html',
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
corider
torchmetrics>=0.4,<0.7
pytorch-lightning>=1.3,<1.5
torchmetrics>=0.4,<0.8
python-dotenv
supers>=0.2
matplotlib
Expand Down
2 changes: 1 addition & 1 deletion ride/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -221,7 +221,7 @@ def profile_model(self, args: AttributeDict, num_runs: int = 100) -> Dict[str, A
elogger.log_metrics(
{
k: v
for k, v in flatten_dict(results, sep="__").items()
for k, v in flatten_dict(results, sep="/").items()
if isinstance(v, Number)
}
)
Expand Down

0 comments on commit 8ea3dc1

Please sign in to comment.