[VERSION] 0.5.1 Pass optional args when initLayer