Spaces:
Runtime error
Runtime error
Update maskrcnn_benchmark/utils/checkpoint.py
Browse files
maskrcnn_benchmark/utils/checkpoint.py
CHANGED
|
@@ -94,6 +94,7 @@ class Checkpointer(object):
|
|
| 94 |
f.write(last_filename)
|
| 95 |
|
| 96 |
def _load_file(self, f):
|
|
|
|
| 97 |
return torch.load(f, map_location=torch.device("cpu"))
|
| 98 |
|
| 99 |
def _load_model(self, checkpoint):
|
|
|
|
| 94 |
f.write(last_filename)
|
| 95 |
|
| 96 |
def _load_file(self, f):
|
| 97 |
+
print(f)
|
| 98 |
return torch.load(f, map_location=torch.device("cpu"))
|
| 99 |
|
| 100 |
def _load_model(self, checkpoint):
|