aryan461 comments

Repositories
Issues
Comments

Results 2 comments of


                                            aryan461

Using the pre-trained model to continue training

I have the same problem and after first training epoch( for all 3 datasets which I checked with `print(len(self.train_dataset))` command and they contain thousands of images) training ends. Does it...

Using the pre-trained model to continue training

@knazeri it was because of `self.iteration = data['iteration']` in [models.py](https://github.com/knazeri/edge-connect/blob/698509d1ac1d7a40310139f9e4d70410b3d734e4/src/models.py#L33). It makes the iteration = 2000000 which is equal to `MAX_ITERS`. I changed the `MAX_ITERS` in config file. Sorry for...