This is a test repository Pathnorm for final layer. Changes introduced - shuffle = False returning two params instead of one from the inference method in resnet.py