fix exp2 pytorch rewrite fatal train issue

2025-11-30 22:01:56 +08:00
parent 48fcdfcc80
commit 43b807679f
13 changed files with 738 additions and 112 deletions
--- a/exp2/source/predict.py
+++ b/exp2/source/predict.py
@@ -5,13 +5,6 @@ import matplotlib.pyplot as plt

 from train import CNN

-'''
-python 3.9 
-tensorflow 2.0.0b0
-pillow(PIL) 4.3.0
-''' 
-
-
 class Predict(object):
    def __init__(self):
        latest = tf.train.latest_checkpoint('./ckpt')