Commit ec808c9b authored by Alan Yee's avatar Alan Yee Committed by Frank Chen
Browse files

Update head.py (#11526)

* Update head.py

Use tf.losses instead of deprecated contrib

* Update head.py

Corrected incorrect import statement

* Update head.py

Correcting shape

* Update head.py

Entered correct type for reduction

* Update head.py

Improve naming convention
parent c0b09bab
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment