| 2 | import argparse |
| 3 | |
| 4 | class TrainCompOptions(BaseOptions): |
| 5 | def initialize(self): |
| 6 | BaseOptions.initialize(self) |
| 7 | self.parser.add_argument('--num_layers', type=int, default=8, help='num_layers of transformer') |
| 8 | self.parser.add_argument('--latent_dim', type=int, default=512, help='latent_dim of transformer') |
| 9 | self.parser.add_argument('--diffusion_steps', type=int, default=1000, help='diffusion_steps of transformer') |
| 10 | self.parser.add_argument('--no_clip', action='store_true', help='whether use clip pretrain') |
| 11 | self.parser.add_argument('--no_eff', action='store_true', help='whether use efficient attention') |
| 12 | |
| 13 | self.parser.add_argument('--num_epochs', type=int, default=50, help='Number of epochs') |
| 14 | self.parser.add_argument('--lr', type=float, default=2e-4, help='Learning rate') |
| 15 | self.parser.add_argument('--batch_size', type=int, default=32, help='Batch size per GPU') |
| 16 | self.parser.add_argument('--times', type=int, default=1, help='times of dataset') |
| 17 | |
| 18 | self.parser.add_argument('--feat_bias', type=float, default=25, help='Scales for global motion features and foot contact') |
| 19 | |
| 20 | self.parser.add_argument('--is_continue', action="store_true", help='Is this trail continued from previous trail?') |
| 21 | |
| 22 | self.parser.add_argument('--log_every', type=int, default=50, help='Frequency of printing training progress (by iteration)') |
| 23 | self.parser.add_argument('--save_every_e', type=int, default=5, help='Frequency of saving models (by epoch)') |
| 24 | self.parser.add_argument('--eval_every_e', type=int, default=5, help='Frequency of animation results (by epoch)') |
| 25 | self.parser.add_argument('--save_latest', type=int, default=500, help='Frequency of saving models (by iteration)') |
| 26 | self.is_train = True |