Commits · f72471dbd68dbccf1f2fd1facf80bfb84603794a · Andrei-Claudiu Roibu / BrainMapper · GitLab

25 Mar, 2020 8 commits
- created a post-training prediction function · f72471db
  Andrei-Claudiu Roibu authored Mar 25, 2020
  
  f72471db
- added test for parameters being on stored on GPU · ea6e15b8
  Andrei-Claudiu Roibu authored Mar 25, 2020
  
  ea6e15b8
- added save function · c2c88a53
  Andrei-Claudiu Roibu authored Mar 25, 2020
  
  c2c88a53
- deleted redundant enable_test_dropout function · 4917a5c2
  Andrei-Claudiu Roibu authored Mar 25, 2020
  
  4917a5c2
- deleted enable_test_dropout - redundant function · 42ae0d12
  Andrei-Claudiu Roibu authored Mar 25, 2020
  
  42ae0d12
- wrote the forward function for U-net · 397a0fc0
  Andrei-Claudiu Roibu authored Mar 25, 2020
  
  397a0fc0
- created constructor function for UNet · 74c4b384
  Andrei-Claudiu Roibu authored Mar 25, 2020
  
  74c4b384
- created the structure for the U-net file · 2f498e71
  Andrei-Claudiu Roibu authored Mar 25, 2020
  
  2f498e71