Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enhancement/train batch function #107

Merged
merged 21 commits into from
Jul 11, 2020
+496 −169
Merged

Enhancement/train batch function #107

Changes from 1 commit
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
9c06583
Updated RL docs with latest models
Jun 24, 2020
33be076
Merge branch 'master' of https://github.com/PyTorchLightning/pytorch-…
Jun 25, 2020
fdc92f9
Merge branch 'master' of https://github.com/PyTorchLightning/pytorch-…
Jun 28, 2020
682bbe6
Merge branch 'master' of https://github.com/PyTorchLightning/pytorch-…
Jun 30, 2020
17073bc
Merge branch 'master' of https://github.com/PyTorchLightning/pytorch-…
Jun 30, 2020
47e5fa0
Merge branch 'master' of https://github.com/PyTorchLightning/pytorch-…
Jul 3, 2020
86b0dee
Added POC for train_batch interface when populating RL datasets
Jul 3, 2020
885f198
Updated other models to use train_batch interface
Jul 3, 2020
896b032
Update tests/datamodules/test_experience_sources.py
djbyrne Jul 6, 2020
2baa02c
Fixing lint errors
Jul 6, 2020
08e71f7
Merge branch 'enhancement/train_batch_function' of https://github.com…
Jul 6, 2020
db18cd8
Fixed linting errors
Jul 6, 2020
0f5ca79
Update pl_bolts/datamodules/experience_source.py
djbyrne Jul 6, 2020
5d9dfa6
Resolved comments
Jul 6, 2020
c3f62ac
req
Borda Jul 8, 2020
577569c
Removed cyclic import of Agents from experience source
Jul 9, 2020
fa658a4
Merge branch 'enhancement/train_batch_function' of https://github.com…
Jul 9, 2020
2292528
Updated reference of Experience to datamodules instead of the rl.common
Jul 9, 2020
13cc727
timeout
Borda Jul 9, 2020
d4c1cc7
Commented out test_dev_dataset to test run times
djbyrne Jul 11, 2020
04f02cd
undo commenting out of test_dev_datasets
djbyrne Jul 11, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Updated reference of Experience to datamodules instead of the rl.common
Donal committed Jul 9, 2020

Verified

This commit was signed with the committer’s verified signature. The key has expired.
danielleadams Danielle Adams
GPG key ID: D3A89613643B6201
Expired
Verified
Learn about vigilant mode
commit 22925289a27138e0120a28a325ea3143c03d45f8
3 changes: 1 addition & 2 deletions tests/datamodules/test_experience_sources.py
View file
Original file line number Diff line number Diff line change
@@ -6,9 +6,8 @@
from torch.utils.data import DataLoader

from pl_bolts.datamodules.experience_source import (ExperienceSourceDataset, ExperienceSource, NStepExperienceSource,
EpisodicExperienceStream)
EpisodicExperienceStream, Experience)
from pl_bolts.models.rl.common.agents import Agent
from pl_bolts.models.rl.common.memory import Experience


class DummyAgent(Agent):