show_model

Author: yanx27

draw_net.py

@@ -1,4 +1,3 @@
-# *_*coding:utf-8 *_*
 import pandas as pd
 import os
 import matplotlib.pyplot as plt
@@ -46,4 +45,4 @@ def show_grapth(img):
 
 from torchviz import make_dot
 draw = make_dot(output, params=dict(model.named_parameters()))
-draw.view()
+draw.view()