Hi, thank you for your great work and the provided pretrained model.
I have some problem when I try to modify the code in the train_loop.py file.
I have total 52001 images and I set 1k image as 1 tick, also I take a snap shot every 1 tick.
Below is where I modified :
total_kimg = 52, # Total length of the training, measured in thousands of real images.kimg_per_tick = 1, # Progress snapshot interval.image_snapshot_ticks = 1, # How often to save image snapshots? None = disable.network_snapshot_ticks = 1, # How often to save network snapshots? None = disable.
And the training script I used is:
python train.py --outdir=output_path --gpus=1 --batch=2 --metrics=fid36k5_full --data=C:/Users/blue/MAT/customdatasets/images.zip --dataloader=datasets.dataset_512.ImageFolderMaskDataset --mirror=True --cond=False --cfg=places512 --aug=noaug --generator=networks.mat.Generator --discriminator=networks.mat.Discriminator --loss=losses.loss.TwoStageLoss --pr=0.1 --pl=False --truncation=0.5 --style_mix=0.5 --ema=10 --lr=0.001 --resume=ffhq512
And here are the files in the output_folder:
It seems that it only saved for the 1st tick. May I ask for some help?
Hi, thank you for your great work and the provided pretrained model. I have some problem when I try to modify the code in the train_loop.py file. I have total 52001 images and I set 1k image as 1 tick, also I take a snap shot every 1 tick. Below is where I modified :
total_kimg = 52, # Total length of the training, measured in thousands of real images.
kimg_per_tick = 1, # Progress snapshot interval.
image_snapshot_ticks = 1, # How often to save image snapshots? None = disable.
network_snapshot_ticks = 1, # How often to save network snapshots? None = disable.
And the training script I used is:
python train.py --outdir=output_path --gpus=1 --batch=2 --metrics=fid36k5_full --data=C:/Users/blue/MAT/customdatasets/images.zip --dataloader=datasets.dataset_512.ImageFolderMaskDataset --mirror=True --cond=False --cfg=places512 --aug=noaug --generator=networks.mat.Generator --discriminator=networks.mat.Discriminator --loss=losses.loss.TwoStageLoss --pr=0.1 --pl=False --truncation=0.5 --style_mix=0.5 --ema=10 --lr=0.001 --resume=ffhq512
And here are the files in the output_folder:![2023-04-08 200240](https://user-images.githubusercontent.com/69571884/230720112-9b361751-c4fe-4363-a173-56d6727efade.png)
It seems that it only saved for the 1st tick. May I ask for some help?