udpate

2023-08-17 11:21:39 +08:00
parent 396cc892d5
commit a56f93d2b5
2 changed files with 1 additions and 2 deletions
--- a/network/BaseNet.py
+++ b/network/BaseNet.py
@ -170,7 +170,6 @@ if __name__ == '__main__':
    img = transform(img)
    img = img.cuda()

-
    # from torchsummary import summary
    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
    model = ResnetFpn().to(device)