i try MCTX Muax its a good art work , but i have issue when i set conv observation like a shape (96, 96, 1) , please guide me to solve this problem
and the correct way to set train_env and eval_env as input of muax.fit() function
Maybe try asking this question on the muax repository: https://github.com/bwfbowen/muax
Try to provide a better description of the problem and maybe include the error message.
i try MCTX Muax its a good art work , but i have issue when i set conv observation like a shape (96, 96, 1) , please guide me to solve this problem and the correct way to set train_env and eval_env as input of muax.fit() function
i use this code :