4be75b3877
Doc String Correction: Attention class forward function
2020-06-15 22:22:51 +05:30
134451267f
Update transformation.py
2020-01-22 03:08:24 +09:00
78d7ed96a9
Update transformation.py
2020-01-17 00:12:18 +09:00
c529003649
Handling breaking change in pytorch grid_sample >=1.4.0 ( https://pytorch.org/docs/stable/nn.functional.html?highlight=grid_sample#torch.nn.functional.grid_sample )
2020-01-16 14:58:59 +01:00
53509eab01
removed align_corners=True
...
Got this error in Pytorch 1.1.0:
TypeError: grid_sample() got an unexpected keyword argument 'align_corners'
2019-10-23 11:35:37 +05:30
f5cf8555ef
updated grid_sample arg align_corners=True
...
UserWarning: Default grid_sample and affine_grid behavior will be changed to align_corners=False from 1.4.0. See the documentation of grid_sample for details.
2019-10-13 20:57:50 +08:00
235095fc9a
Update transformation.py
2019-09-27 18:06:16 +09:00
8f7255fb4b
.cuda() to .to(device)
2019-08-03 08:03:46 +00:00
3772d75f40
Update prediction.py
2019-04-26 14:27:34 +09:00
c2e28f5c0d
comment and multi-gpu setting update
2019-04-16 07:28:53 +00:00
cb254fa63a
GridGenerator update
2019-04-15 09:41:24 +00:00
81d381701f
fix minor
2019-04-12 11:59:19 +00:00
7553b47a7c
register buffer in TPS-grid for multi-gpu
2019-04-11 09:44:20 +00:00
7da666fc3a
update model arg
2019-04-09 09:06:32 +00:00
34e6316856
fix minor
2019-04-07 12:41:11 +00:00
2b38eb2574
'init'
2019-04-05 19:45:29 +09:00
Tejas Mahajan