Description
The current NumPy-only implementation of convac / CNN-LA is somewhat difficult to maintain because of manual gradient computation. It also does not support GPUs. This story is to re-implement this network architecture in pytorch with optional GPU training support.
Acceptance Criteria
Tasks
Description
The current NumPy-only implementation of
convac/ CNN-LA is somewhat difficult to maintain because of manual gradient computation. It also does not support GPUs. This story is to re-implement this network architecture inpytorchwith optional GPU training support.Acceptance Criteria
convacavailable and written inpytorchconvacimplementation supports both CPU and GPU trainingTasks