train.py 244 Bytes Raw Blame History Permalink 1 2 3 4 5 6 7 8 9 10 11 from model import DTN from solver import Solver def main(): model = DTN() solver = Solver(model, num_epoch=10, svhn_path='svhn/', model_save_path='model/', log_path='log/') solver.train() if __name__ == "__main__": main()