Currently the system uses 2 Training files for 2 phases of training. This can cause some unwanted future issues and it would be really difficult to track the model.
The Following Updates are requested,
Is this code structure of having a different file per-training phase followed elsewhere? Instead you can split your original model file into one module each for G, D & the overall model. And add each training phase as a helper on the main model class - this way its easier to track the behavior of the model across its entire lifetime in the pipeline. Originally posted by @srjoglekar246 in https://github.com/captain-pool/GSOC/pull/28
The idea is to do the following:
[x] Update the training codes for phase_1 and phase_2 and convert the class based approach to trainer functions.
[x] migrate all the training codes from both the phases to one file.
Currently the system uses 2 Training files for 2 phases of training. This can cause some unwanted future issues and it would be really difficult to track the model. The Following Updates are requested,
The idea is to do the following:
phase_1andphase_2and convert the class based approach to trainer functions.