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

fixing TPU tests #2632

Merged
merged 55 commits into from
Jul 27, 2020
Merged

fixing TPU tests #2632

Changes from 1 commit
Commits
Show all changes
55 commits
Select commit Hold shift + click to select a range
bacc208
init
Borda Jul 17, 2020
b93fab4
rename
Borda Jul 17, 2020
15cb65c
tpu_core_idx
Borda Jul 17, 2020
5ac3c61
idx 8
Borda Jul 17, 2020
724b7b4
idxs
Borda Jul 17, 2020
d817f0e
@pl_multi_process_test
Borda Jul 17, 2020
be79c50
assert
Borda Jul 17, 2020
f35437f
assert
Borda Jul 17, 2020
773e52a
deamon
Borda Jul 17, 2020
30cda41
no close
Borda Jul 17, 2020
be98712
imort
Borda Jul 18, 2020
f93c5aa
msg
Borda Jul 19, 2020
1dab0c8
use_single_gpu
Borda Jul 21, 2020
910f1c4
dataset
Borda Jul 21, 2020
410742a
idx
Borda Jul 21, 2020
2ad7b42
fix idx
Borda Jul 22, 2020
7576549
dataset
Borda Jul 22, 2020
accea7f
format
Borda Jul 23, 2020
85759bd
add pickable
Borda Jul 23, 2020
c8e7a70
typo
Borda Jul 25, 2020
19919a3
apex
Borda Jul 25, 2020
4ed34a6
typo
Borda Jul 25, 2020
b11c6bc
wip
Borda Jul 27, 2020
dee7fa5
wip
Borda Jul 27, 2020
e2ece1b
wip
Borda Jul 27, 2020
6c11dae
wip
Borda Jul 27, 2020
c216283
wip
Borda Jul 27, 2020
4f726a7
wip
Borda Jul 27, 2020
91d1f56
wip
Borda Jul 27, 2020
d6f1137
wip
Borda Jul 27, 2020
9465404
docs
Borda Jul 27, 2020
75dc5d2
typo
Borda Jul 27, 2020
5a7bfa0
tests
Borda Jul 27, 2020
22dd8c0
tests
Borda Jul 27, 2020
e4f2088
tests
Borda Jul 27, 2020
d89de9b
tests
Borda Jul 27, 2020
1614724
tests
Borda Jul 27, 2020
def846b
tests
Borda Jul 27, 2020
f6060ec
tests
Borda Jul 27, 2020
f0a6174
tests
Borda Jul 27, 2020
bdaaaef
tests
Borda Jul 27, 2020
458471d
tests
Borda Jul 27, 2020
6e6588a
tests
Borda Jul 27, 2020
6dc0e85
tests
Borda Jul 27, 2020
6196724
tests
Borda Jul 27, 2020
183fcc2
tests
Borda Jul 27, 2020
e7a5295
tests
Borda Jul 27, 2020
1999198
tests
Borda Jul 27, 2020
9579bce
tests
Borda Jul 27, 2020
fde01de
docs
Borda Jul 27, 2020
6abee20
docs
Borda Jul 27, 2020
bf6ac74
Apply suggestions from code review
Borda Jul 27, 2020
6f30bc5
Apply suggestions from code review
Borda Jul 27, 2020
45233a5
docs
Borda Jul 27, 2020
6c9495e
Apply suggestions from code review
Borda Jul 27, 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
wip
  • Loading branch information
@Borda
Borda committed Jul 27, 2020
commit c21628399995cd69cfcbca1cbaddfe1d3549f93a
6 changes: 3 additions & 3 deletions pytorch_lightning/accelerator_backends/tpu_backend.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,9 +63,9 @@ def teardown(self, model):
# todo, pass also bets score

# load last weights
if last_path is not None and not self.trainer.testing:
ckpt = torch.load(last_path, map_location=lambda storage, loc: storage)
model.load_state_dict(ckpt)
# if last_path is not None and not self.trainer.testing:
# ckpt = torch.load(last_path, map_location=lambda storage, loc: storage)
# model.load_state_dict(ckpt)

self.trainer.model = model

Expand Down