search

snwagh / falcon-public

Implementation of protocols in Falcon
90 stars 46 forks source link

Revert inputDim and batchSize #27

Closed yl3469 closed 2 years ago

yl3469 commented 2 years ago

The order of inputDim and batchSize is defined in ReLUCOnfig as follows. We should preserve the order when using it. :) 

class ReLUConfig : public LayerConfig
{
public:
    size_t inputDim = 0;
    size_t batchSize = 0;

    ReLUConfig(size_t _inputDim, size_t _batchSize)
    :inputDim(_inputDim),
     batchSize(_batchSize), 
     LayerConfig("ReLU")
    {};
};
snwagh commented 2 years ago

Thanks for spotting this, merging it now.