Custom architecture

[abhigoku10]

@mightydeveloper hi thanks for the wonderful code base , i have following few queries
1.Can we reduce the weight size of the model with the following code base
2. can your code base be applied to custom architecture

[samohadid]

hello according to the original paper this method can be applied to any custom architecture