Commit d3cac54

authored

Update main.py

totalEpoch =len(loader.trainSamples)//Model.batchSize gets changed.

1 parent 4e142d4 commit d3cac54Copy full SHA for d3cac54

File tree

-1

lines changed

-1

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,7 @@ def train(model, loader):`
`21`	`21`	`earlyStopping = 25 # Stop training after this number of epochs without improvement`
`22`	`22`	`batchNum = 0`
`23`	`23`
`24`		`- totalEpoch = loader.trainSamples//Model.batchSize # loader.numTrainSamplesPerEpoch`
	`24`	`+ totalEpoch = len(loader.trainSamples)//Model.batchSize`
`25`	`25`
`26`	`26`	`while True:`
`27`	`27`	`epoch += 1`

Comments

(0)