diff --git a/docs/build/doctrees/environment.pickle b/docs/build/doctrees/environment.pickle
index 3b50fc6..66dc3ec 100644
Binary files a/docs/build/doctrees/environment.pickle and b/docs/build/doctrees/environment.pickle differ
diff --git a/docs/build/doctrees/index.doctree b/docs/build/doctrees/index.doctree
index deca016..e72d14b 100644
Binary files a/docs/build/doctrees/index.doctree and b/docs/build/doctrees/index.doctree differ
diff --git a/docs/build/html/index.html b/docs/build/html/index.html
index 7486540..877b048 100644
--- a/docs/build/html/index.html
+++ b/docs/build/html/index.html
@@ -212,7 +212,7 @@
Description
Import the package and Instantiate any module/block class you want to you:
from nn_common_modules import modules as additional_modules
-loss = additional_modules.DenseBlock(params, se_block_type = 'SSE')
+dense_block = additional_modules.DenseBlock(params, se_block_type = 'SSE')
diff --git a/docs/build/html/searchindex.js b/docs/build/html/searchindex.js
index ab72f01..e720150 100644
--- a/docs/build/html/searchindex.js
+++ b/docs/build/html/searchindex.js
@@ -1 +1 @@
-Search.setIndex({docnames:["index"],envversion:53,filenames:["index.rst"],objects:{"":{losses:[0,0,0,"-"],modules:[0,0,0,"-"]},"losses.CombinedLoss":{forward:[0,2,1,""]},"losses.CrossEntropyLoss2d":{forward:[0,2,1,""]},"losses.DiceLoss":{forward:[0,2,1,""]},"losses.FocalLoss":{forward:[0,2,1,""]},"losses.IoULoss":{forward:[0,2,1,""]},"modules.ClassifierBlock":{forward:[0,2,1,""]},"modules.DecoderBlock":{forward:[0,2,1,""]},"modules.DenseBlock":{forward:[0,2,1,""]},"modules.EncoderBlock":{forward:[0,2,1,""]},"modules.GenericBlock":{forward:[0,2,1,""]},"modules.SDnetDecoderBlock":{forward:[0,2,1,""]},"modules.SDnetEncoderBlock":{forward:[0,2,1,""]},losses:{CombinedLoss:[0,1,1,""],CrossEntropyLoss2d:[0,1,1,""],DiceLoss:[0,1,1,""],FocalLoss:[0,1,1,""],IoULoss:[0,1,1,""]},modules:{ClassifierBlock:[0,1,1,""],DecoderBlock:[0,1,1,""],DenseBlock:[0,1,1,""],EncoderBlock:[0,1,1,""],GenericBlock:[0,1,1,""],SDnetDecoderBlock:[0,1,1,""],SDnetEncoderBlock:[0,1,1,""]}},objnames:{"0":["py","module","Python module"],"1":["py","class","Python class"],"2":["py","method","Python method"]},objtypes:{"0":"py:module","1":"py:class","2":"py:method"},terms:{"class":0,"default":0,"function":[],"import":0,"int":0,"return":0,"true":0,"while":[],_build:[],addit:0,additional_loss:0,additional_modul:0,afterward:[],all:[],alpha:0,although:[],ani:0,api:[],argument:[],batch:0,batchnorm:0,binar:0,binari:0,block:0,bool:0,build:0,call:[],can:0,care:[],chaneel:0,classifi:0,classifierblock:0,classs:0,combin:0,combinedloss:0,commonli:0,comput:[],connect:0,conv:0,convolut:0,cross:0,crossentropyloss2d:0,crossentropyloss:0,cse:0,csse:0,custom:0,decod:0,decoderblock:0,deep:0,defin:0,dens:0,denseblock:0,depend:0,detect:0,dice:0,diceloss:0,dict:0,drop_out:0,encod:0,encoderblock:0,entropi:0,everi:[],excit:0,fals:0,few:[],floattensor:0,focal:0,focalloss:0,former:[],forward:0,from:0,gamma:0,gener:0,genericblock:0,github:[],hook:[],html:[],http:[],ignor:0,ignore_index:0,includ:0,index:0,indic:0,input:0,instanc:[],instanti:0,instead:[],iou:0,iouloss:0,kernel_c:0,kernel_h:0,kernel_w:0,last:0,latter:[],layer:0,learn:[],librari:0,logit:0,longtensor:0,loss:[],manual:[],maxpool:0,maxunpool:0,mean:0,need:[],netwok:0,network:0,neural:0,nn_common_modul:0,none:0,num_channel:0,num_class:0,num_filt:0,nxc:0,nxcxhxw:0,nxhxw:0,object:0,one:0,oper:0,option:0,out_block:0,output:0,overridden:[],packag:0,page:[],param:0,paramet:0,parent:0,part:0,particularli:0,pass:0,perform:[],pleas:[],pool:0,prelu:0,pytorch:0,recip:[],reduc:0,reduct:0,refer:[],regist:[],regress:0,relu:[],reus:0,rtype:[],run:[],sampl:0,scalar:0,sdnetdecoderblock:0,sdnetencoderblock:0,se_block:0,se_block_typ:0,search:[],segment:0,selay:0,shape:0,shayansiddiqui:[],shot:[],should:[],silent:[],sinc:[],size_averag:0,skip:0,skipconnect:0,spatial:[],specif:0,squeez:0,sse:0,standard:0,str:0,stride_conv:0,stride_pool:0,subclass:[],summari:[],take:[],target:0,technic:[],tensor:0,them:[],thi:[],tonsor:0,torch:0,tweak:[],type:0,unpool:0,usag:[],use:[],used:0,valid:0,valu:0,variabl:[],want:0,weight:0,which:0,within:[],without:0,you:0},titles:["nn-common-modules\u2019s documentation!"],titleterms:{common:0,descript:0,document:0,indic:[],loss:0,member:0,modul:0,tabl:[],usag:0,welcom:[]}})
\ No newline at end of file
+Search.setIndex({docnames:["index"],envversion:53,filenames:["index.rst"],objects:{"":{losses:[0,0,0,"-"],modules:[0,0,0,"-"]},"losses.CombinedLoss":{forward:[0,2,1,""]},"losses.CrossEntropyLoss2d":{forward:[0,2,1,""]},"losses.DiceLoss":{forward:[0,2,1,""]},"losses.FocalLoss":{forward:[0,2,1,""]},"losses.IoULoss":{forward:[0,2,1,""]},"modules.ClassifierBlock":{forward:[0,2,1,""]},"modules.DecoderBlock":{forward:[0,2,1,""]},"modules.DenseBlock":{forward:[0,2,1,""]},"modules.EncoderBlock":{forward:[0,2,1,""]},"modules.GenericBlock":{forward:[0,2,1,""]},"modules.SDnetDecoderBlock":{forward:[0,2,1,""]},"modules.SDnetEncoderBlock":{forward:[0,2,1,""]},losses:{CombinedLoss:[0,1,1,""],CrossEntropyLoss2d:[0,1,1,""],DiceLoss:[0,1,1,""],FocalLoss:[0,1,1,""],IoULoss:[0,1,1,""]},modules:{ClassifierBlock:[0,1,1,""],DecoderBlock:[0,1,1,""],DenseBlock:[0,1,1,""],EncoderBlock:[0,1,1,""],GenericBlock:[0,1,1,""],SDnetDecoderBlock:[0,1,1,""],SDnetEncoderBlock:[0,1,1,""]}},objnames:{"0":["py","module","Python module"],"1":["py","class","Python class"],"2":["py","method","Python method"]},objtypes:{"0":"py:module","1":"py:class","2":"py:method"},terms:{"class":0,"default":0,"function":[],"import":0,"int":0,"return":0,"true":0,"while":[],_build:[],addit:0,additional_loss:0,additional_modul:0,afterward:[],all:[],alpha:0,although:[],ani:0,api:[],argument:[],batch:0,batchnorm:0,binar:0,binari:0,block:0,bool:0,build:0,call:[],can:0,care:[],chaneel:0,classifi:0,classifierblock:0,classs:0,combin:0,combinedloss:0,commonli:0,comput:[],connect:0,conv:0,convolut:0,cross:0,crossentropyloss2d:0,crossentropyloss:0,cse:0,csse:0,custom:0,decod:0,decoderblock:0,deep:0,defin:0,dens:0,dense_block:0,denseblock:0,depend:0,detect:0,dice:0,diceloss:0,dict:0,drop_out:0,encod:0,encoderblock:0,entropi:0,everi:[],excit:0,fals:0,few:[],floattensor:0,focal:0,focalloss:0,former:[],forward:0,from:0,gamma:0,gener:0,genericblock:0,github:[],hook:[],html:[],http:[],ignor:0,ignore_index:0,includ:0,index:0,indic:0,input:0,instanc:[],instanti:0,instead:[],iou:0,iouloss:0,kernel_c:0,kernel_h:0,kernel_w:0,last:0,latter:[],layer:0,learn:[],librari:0,logit:0,longtensor:0,loss:[],manual:[],maxpool:0,maxunpool:0,mean:0,need:[],netwok:0,network:0,neural:0,nn_common_modul:0,none:0,num_channel:0,num_class:0,num_filt:0,nxc:0,nxcxhxw:0,nxhxw:0,object:0,one:0,oper:0,option:0,out_block:0,output:0,overridden:[],packag:0,page:[],param:0,paramet:0,parent:0,part:0,particularli:0,pass:0,perform:[],pleas:[],pool:0,prelu:0,pytorch:0,recip:[],reduc:0,reduct:0,refer:[],regist:[],regress:0,relu:[],reus:0,rtype:[],run:[],sampl:0,scalar:0,sdnetdecoderblock:0,sdnetencoderblock:0,se_block:0,se_block_typ:0,search:[],segment:0,selay:0,shape:0,shayansiddiqui:[],shot:[],should:[],silent:[],sinc:[],size_averag:0,skip:0,skipconnect:0,spatial:[],specif:0,squeez:0,sse:0,standard:0,str:0,stride_conv:0,stride_pool:0,subclass:[],summari:[],take:[],target:0,technic:[],tensor:0,them:[],thi:[],tonsor:0,torch:0,tweak:[],type:0,unpool:0,usag:[],use:[],used:0,valid:0,valu:0,variabl:[],want:0,weight:0,which:0,within:[],without:0,you:0},titles:["nn-common-modules\u2019s documentation!"],titleterms:{common:0,descript:0,document:0,indic:[],loss:0,member:0,modul:0,tabl:[],usag:0,welcom:[]}})
\ No newline at end of file
diff --git a/nn_common_modules/modules.py b/nn_common_modules/modules.py
index b3c1d89..c2817f0 100644
--- a/nn_common_modules/modules.py
+++ b/nn_common_modules/modules.py
@@ -8,7 +8,7 @@
Import the package and Instantiate any module/block class you want to you::
from nn_common_modules import modules as additional_modules
- loss = additional_modules.DenseBlock(params, se_block_type = 'SSE')
+ dense_block = additional_modules.DenseBlock(params, se_block_type = 'SSE')
Members
++++++++++++++++++++++