search

echonet / dynamic

EchoNet-Dynamic is a deep learning model for assessing cardiac function in echocardiogram videos.
https://echonet.github.io/dynamic
Other
482 stars 169 forks source link

Test in initialization (TypeError: enabled must be a bool (got str)) #38

Open gt-madanb opened 3 years ago

gt-madanb commented 3 years ago

0%| | 0/4 [00:00<?, ?it/s] loading weights from D:\stanford_AIMI\weights\r2plus1d_18_32_2_pretrained cuda is not available, cpu weights EXTERNAL_TEST ['0X1A030EFDD45062FA.avi', '0X1A05DFFFCAFB253B.avi', '0X1A09BE7969DA1508.avi', '0X1A0A263B22CCD966.avi', '0X1A193A8138F4DD0F.avi', '0X1A296F5FCD5A0ED8.avi', '0X1A2A76BDB5B98BED.avi', '0X1A2C60147AF9FDAE.avi', '0X1A2E9496910EFF5B.avi', '0X1A349D84388BD74B.avi', '0X1A36AE0874972BBE.avi', '0X1A3D565B371DC573.avi', '0X1A3E7BF1DFB132FB.avi', '0X1A481BE9AE4F2DCD.avi', '0X1A494FC3B214947B.avi', '0X1A58B506ED05C1D4.avi', '0X1A58C9DFE12C7953.avi', '0X1A5FAE3F9D37794E.avi', '0X1A62D321A1A1821B.avi', '0X1A6ACFE7B286DAFC.avi', '0X1A75CDAE16981DBC.avi', '0X1A76A1A8448B456.avi', '0X1A8D85542DBE8204.avi', '0X1A8F20B8BF0B4B45.avi', '0X1A970F020826E7FA.avi', '0X1A9D7251E9464D49.avi'] 100%|████████████████████████████████████████████████████████████████████████████████████| 4/4 [00:03<00:00, 1.12it/s]

TypeError Traceback (most recent call last)

in 39 ds.crops="all" 40 test_dataloader = torch.utils.data.DataLoader(ds, batch_size = 1, num_workers = 5, shuffle = True, pin_memory=(device.type == "cuda")) ---> 41 loss, yhat, y = echonet.utils.video.run_epoch(model, test_dataloader, "test", None, device)#, save_all=True)#, blocks=25) 42 43 with open(output, "w") as g: ~\dynamic-master\echonet\utils\video.py in run_epoch(model, dataloader, train, optim, device, save_all, block_size) 312 y = [] 313 --> 314 with torch.set_grad_enabled(train): 315 with tqdm.tqdm(total=len(dataloader)) as pbar: 316 for (X, outcome) in dataloader: ~\anaconda3\lib\site-packages\torch\autograd\grad_mode.py in __init__(self, mode) 200 def __init__(self, mode: bool) -> None: 201 self.prev = torch.is_grad_enabled() --> 202 torch._C._set_grad_enabled(mode) 203 204 def __enter__(self) -> None: TypeError: enabled must be a bool (got str)