Closed zihxiao closed 5 years ago
delete batch size in initial, add parameters epochs and batch_size in getBatch() and fit(), change the tensor names in fit(), change self.training into a tensor
delete batch size in initial, add parameters epochs and batch_size in getBatch() and fit(), change the tensor names in fit(), change self.training into a tensor