changed hps

td3/main.py

@@ -27,11 +27,11 @@
 parser.add_argument('--img_input', action="store_true", default=False, help='Use image as states')
 parser.add_argument('--in_channels', type=int, default=3, help='Number of image channels for image input')
 parser.add_argument('--depth', type=int, default=3, help='Depth for CNN architecture for image input')
-parser.add_argument('--multiplier', type=int, default=64, help='Channel multiplier for CNN architecture for image input')
+parser.add_argument('--multiplier', type=int, default=32, help='Channel multiplier for CNN architecture for image input')
 parser.add_argument('--order', type=int, default=3, help='Store past (order) of frames for image input')
 parser.add_argument('--action_embed_dim', type=int, default=32, help='Embedding dimension for actions for image input')
-parser.add_argument('--hidden_dim', type=list, default=1028, help='List of hidden dims for embedding networks')
-parser.add_argument('--crop_dim', type=list, default=32, help='Crop dim for image inputs')
+parser.add_argument('--hidden_dim', type=int, default=512, help='List of hidden dims for embedding networks')
+parser.add_argument('--crop_dim', type=int, default=32, help='Crop dim for image inputs')
 
 # training hp params
 parser.add_argument('--n_episodes', type=int, default=1000, help='Number of episodes')