-
Notifications
You must be signed in to change notification settings - Fork 1
Open
Description
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
- New implementation of
convacavailable and written inpytorch - New
convacimplementation supports both CPU and GPU training
Tasks
- create new module structure for torch re-implementations
- implement convac
- add GPU support
- add unit tests
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels