a lot of training

This commit is contained in:
2023-03-01 19:31:57 +00:00
parent 35ba014ba5
commit 424f8d2fcc
20 changed files with 1365 additions and 27 deletions
+2 -2
View File
@@ -1,6 +1,6 @@
import os
import sys
from configs.iccv.config_abl_skin import config
from configs.iccv.config_abl_skin_reg_skirt import config
os.environ["CUDA_VISIBLE_DEVICES"] = config['gpu']
import tensorflow as tf
@@ -546,7 +546,7 @@ if __name__ == '__main__':
train_loss = train_step(batch)
train_accumulator.update(train_loss)
if step % 50 == 0:
if step % 25 == 0:
data_valid = valid_iterator.get_next()
data_valid = static_dataset_valid.expand_batch(batch_to_dict(data_valid, config['signature_dynamic']))
valid_loss = valid_step(data_valid)