(states, is_best, output_dir,
filename='checkpoint.pth')
| 82 | return path_dict |
| 83 | |
| 84 | def save_checkpoint(states, is_best, output_dir, |
| 85 | filename='checkpoint.pth'): |
| 86 | torch.save(states, os.path.join(output_dir, filename)) |
| 87 | if is_best: |
| 88 | torch.save(states, os.path.join(output_dir, 'checkpoint_best.pth')) |
| 89 | |
| 90 | def random_click(mask, point_labels = 1, seed=None): |
| 91 | # check if all masks are black |
nothing calls this directly
no outgoing calls
no test coverage detected