Skip to content

Commit

Permalink
Randomly initialize AlexNet input
Browse files Browse the repository at this point in the history
partially addresses issue #8
  • Loading branch information
patflick committed Jul 19, 2018
1 parent 1de10c1 commit eebcfca
Showing 1 changed file with 2 additions and 9 deletions.
11 changes: 2 additions & 9 deletions alexnet.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -5,15 +5,7 @@
#include "multi_layers.hpp"


/* TODO:
* - [ ] create AlexNet class
* - [ ] uniform random tensors (via host->device copy), and CPU initialized tensors
* - [x] Make `Model` take input and output tensors in forward(), backward()
* - [ ] Collect total and average times per layer
* - [ ] implement and benchmark ResNet

This comment has been minimized.

Copy link
@mattsinc

mattsinc Jul 19, 2018

@patflick you can probably check this off your list too :)

This comment has been minimized.

Copy link
@patflick

patflick Jul 19, 2018

Author Owner

yeah, that's why i deleted it all =)

*/


// implemenets AlexNet

This comment has been minimized.

Copy link
@mattsinc

mattsinc Jul 19, 2018

@patflick typo

void alexNet() {
TensorDesc input_dim(128, 3, 224, 224);

Expand Down Expand Up @@ -49,6 +41,7 @@ void alexNet() {
Model m(input_dim);
m.add(features);
m.add(classifier);
m.input.uniform(); // randomly initialize input

BenchmarkLogger::new_session("alex_net");
BenchmarkLogger::benchmark(m, 50);
Expand Down

0 comments on commit eebcfca

Please sign in to comment.