We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 6e4bbb7 commit 0f83affCopy full SHA for 0f83aff
research/ptn/train_ptn.py
@@ -44,7 +44,7 @@
44
'Input images dimension (pixels) - width & height.')
45
flags.DEFINE_integer('vox_size', 32, 'Voxel prediction dimension.')
46
flags.DEFINE_integer('step_size', 24, 'Steps to take in rotation to fetch viewpoints.')
47
-flags.DEFINE_integer('batch_size', 1, 'Batch size while training.')
+flags.DEFINE_integer('batch_size', 6, 'Batch size while training.')
48
flags.DEFINE_float('focal_length', 0.866, 'Focal length parameter used in perspective projection.')
49
flags.DEFINE_float('focal_range', 1.732, 'Focal length parameter used in perspective projection.')
50
flags.DEFINE_string('encoder_name', 'ptn_encoder',
0 commit comments