mirror of https://github.com/commaai/tinygrad.git
Fix plt output comment (#1428)
This commit is contained in:
parent
801bed4f66
commit
c08ed1949f
|
@ -50,7 +50,7 @@ def infer(model, img):
|
|||
# if you want to look at the outputs
|
||||
"""
|
||||
import matplotlib.pyplot as plt
|
||||
plt.plot(out.numpy()[0])
|
||||
plt.plot(out[0])
|
||||
plt.show()
|
||||
"""
|
||||
return out, retimg
|
||||
|
|
Loading…
Reference in New Issue