Which pytorch version do I need?

[harsha_g]

@S_Ellis

AttributeError: ‘BatchNorm2d’ object has no attribute ‘track_running_stats’

How about taking the middle road? Did you try 0.3.1? Got it from here.